MCPcopy Create free account
hub / github.com/NVIDIA/SOL-ExecBench / test_float16

Method test_float16

tests/sol_execbench/core/bench/test_io.py:54–56  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

52 t = _rand_tensor([4, 8], torch.float32, torch.device("cpu"))
53 assert t.shape == (4, 8)
54 assert t.dtype == torch.float32
55
56 def test_float16(self):
57 t = _rand_tensor([3], torch.float16, torch.device("cpu"))
58 assert t.dtype == torch.float16
59

Callers

nothing calls this directly

Calls 1

_rand_tensorFunction · 0.90

Tested by

no test coverage detected