(x: tvm_ffi.Tensor)
| 192 | |
| 193 | @tvm_ffi.register_global_func("testing.check_device", override=True) |
| 194 | def _check_device(x: tvm_ffi.Tensor) -> str: |
| 195 | return x.device.type |
| 196 | |
| 197 | # PyTorch uses device name "cuda" to represent ROCm device |
| 198 | x = torch.randn(128, device="cuda") |
nothing calls this directly
no outgoing calls
no test coverage detected