Function _dist_is_main

dflash/benchmark.py:158–159 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

156
157
158	def _dist_is_main() -> bool:
159	return _dist_rank() == 0
160
161
162	def _dist_gather(torch_dist, obj: Any, dst: int = 0):

_dist_gatherFunction · 0.85

_run_transformersFunction · 0.85

_dist_rankFunction · 0.85

no test coverage detected