add back "CPU" in test_onnx_backend supports_device (#3426)

the onnx tests were all skipped.
This commit is contained in:
chenyu
2024-02-16 00:49:30 -05:00
committed by GitHub
parent 28a8b72024
commit 30f26279c5
4 changed files with 3 additions and 6 deletions

View File

@@ -414,7 +414,6 @@ class TestZeroShapeTensor(unittest.TestCase):
assert t.shape == (3, 2, 2)
np.testing.assert_equal(t.numpy(), s.numpy())
# torch does not support padding non-zero dim with 0-size. torch.nn.functional.pad(torch.zeros(3,2,0), [0,0,0,4,0,0])
s = Tensor.rand(3, 4, 0)
t = Tensor.rand(3, 2, 0).cat(s, dim=1)
assert t.shape == (3, 6, 0)