Function _is_supported

axlearn/common/host_array_test.py:58–59 · view source on GitHub ↗

(*, platform: str, mesh_shape: MeshShape)

Source from the content-addressed store, hash-verified

56
57
58	def _is_supported(*, platform: str, mesh_shape: MeshShape) -> bool:
59	return is_supported_platform(platform) and is_supported_mesh_shape(mesh_shape)
60
61
62	def _ordered_devices(mesh_shape: MeshShape, process_shape: MeshShape) -> np.ndarray:

test_fixed_process_shapeMethod · 0.85

test_fixed_global_shapeMethod · 0.85

test_host_to_global_multiple_dimsMethod · 0.85

test_global_to_host_multiple_dimsMethod · 0.85

is_supported_platformFunction · 0.90

is_supported_mesh_shapeFunction · 0.90

no test coverage detected