(idx)
| 116 | |
| 117 | def test_tensor_cpu_from_dlpack(): |
| 118 | def create_tmp(idx): |
| 119 | a = np.full((4, 4), idx) |
| 120 | dlt = to_dlpack(torch.from_numpy(a)) |
| 121 | return tensors.TensorCPU(dlt, "") |
| 122 | |
| 123 | out = [create_tmp(i) for i in range(4)] |
| 124 | for i, t in enumerate(out): |
no outgoing calls
no test coverage detected