MCPcopy Index your code
hub / github.com/FlashSampling/FlashSampling / verify_correctness_tp2

Function verify_correctness_tp2

src/fused_mm_sampling/testing.py:162–170  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

160
161
162def verify_correctness_tp2() -> None:
163 tp = TPInfo.from_world()
164 tp.rank0_print("=== test_sampling_distribution_tp2 ===")
165 verify_sampling_distribution_tp2()
166
167 tp.rank0_print("\n=== test_greedy_tp2 ===")
168 verify_greedy_tp2()
169
170 tp.rank0_print("\nAll distributed tests passed.")
171
172
173def verify_sampling_distribution_tp2() -> None:

Callers

nothing calls this directly

Calls 4

verify_greedy_tp2Function · 0.85
from_worldMethod · 0.80
rank0_printMethod · 0.80

Tested by

no test coverage detected