Function get_world_size

tests/test_update.py:26–27 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

24
25
26	def get_world_size() -> int:
27	return device_manager.device_module.device_count()
28
29
30	def gen_test_tensors(rank: int) -> list[tuple[str, torch.Tensor]]:

get_filesFunction · 0.90

run_with_filesFunction · 0.85

no outgoing calls

no test coverage detected