MCPcopy Create free account
hub / github.com/apple/axlearn / MockUtilizationSample

Class MockUtilizationSample

axlearn/common/monitoring/gpu_client_test.py:23–24  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

21
22@dataclasses.dataclass
23class MockUtilizationSample:
24 sampleValue: MockUtilizationSampleValue # pylint: disable=invalid-name
25
26
27@unittest.skipIf(jax.default_backend() != "gpu", "Skip due to not a gpu backend")

Callers 2

test_get_gpu_device_utilizationMethod · 0.85
test_get_gpu_device_memory_utilizationMethod · 0.85

Calls

no outgoing calls

Tested by 2

test_get_gpu_device_utilizationMethod · 0.68
test_get_gpu_device_memory_utilizationMethod · 0.68