已合并
skip ci error #31490
SCh_zx创建于 3月6日
skip ci error #31490
已合并
SCh_zx创建于 3月6日
1 个文件变更+1-1
@@ -8,7 +8,7 @@ from torch_npu.testing.decorator import Dtypes, instantiate_tests
8@instantiate_tests8@instantiate_tests
9class TestDLPack(TestCase):9class TestDLPack(TestCase):
10 10 
11- @Dtypes(torch.float, torch.double, torch.int8, torch.int16, torch.int32, torch.int64, torch.uint8, torch.bool)11+ @Dtypes(torch.float, torch.double, torch.int8, torch.int16, torch.int32, torch.int64, torch.bool)
12 def test_dlpack_roundtrip_basic(self, dtype, device="npu"):12 def test_dlpack_roundtrip_basic(self, dtype, device="npu"):
13 """Test basic dlpack roundtrip: torch_npu tensor -> dlpack -> torch_npu tensor"""13 """Test basic dlpack roundtrip: torch_npu tensor -> dlpack -> torch_npu tensor"""
14 # Create original tensor14 # Create original tensor