已合并
[test] skip uniform ut in A5 #5188
wanglijun55创建于 6月15日
[test] skip uniform ut in A5 #5188
已合并
共 1 个文件变更+2-0
| @@ -2,6 +2,7 @@ import torch | |||
| 2 | import torch_npu | 2 | import torch_npu |
| 3 | 3 | ||
| 4 | from torch_npu.testing.testcase import TestCase, run_tests | 4 | from torch_npu.testing.testcase import TestCase, run_tests |
| 5 | +from torch_npu.testing.common_utils import SupportedDevices | ||
| 5 | 6 | ||
| 6 | 7 | ||
| 7 | class TestUniform(TestCase): | 8 | class TestUniform(TestCase): |
| @@ -17,6 +18,7 @@ class TestUniform(TestCase): | |||
| 17 | self.assertTrue(item[1] <= input1.min()) | 18 | self.assertTrue(item[1] <= input1.min()) |
| 18 | self.assertTrue(item[2] >= input1.max()) | 19 | self.assertTrue(item[2] >= input1.max()) |
| 19 | 20 | ||
| 21 | + | ||
| 20 | def test_uniform_trans(self): | 22 | def test_uniform_trans(self): |
| 21 | shape_format = [ | 23 | shape_format = [ |
| 22 | [(20, 300), -100, 100, torch.float32], | 24 | [(20, 300), -100, 100, torch.float32], |