已合并
skip ci error (update cann) #34621
SCh_zx创建于 4月28日
skip ci error (update cann) #34621
已合并
共 1 个文件变更+3-0
| @@ -1,4 +1,5 @@ | |||
| 1 | import re | 1 | import re |
| 2 | +import unittest | ||
| 2 | 3 | ||
| 3 | from torch.testing._internal.common_utils import TestCase, run_tests | 4 | from torch.testing._internal.common_utils import TestCase, run_tests |
| 4 | import torch | 5 | import torch |
| @@ -8,6 +9,7 @@ from torch_npu.npu.utils import get_cann_version, _is_gte_cann_version | |||
| 8 | 9 | ||
| 9 | 10 | ||
| 10 | class TestCANNversion(TestCase): | 11 | class TestCANNversion(TestCase): |
| 12 | + | ||
| 11 | def test_get_cann_version(self): | 13 | def test_get_cann_version(self): |
| 12 | version_env = get_cann_version_from_env() | 14 | version_env = get_cann_version_from_env() |
| 13 | version = get_cann_version(module="CANN") | 15 | version = get_cann_version(module="CANN") |
| @@ -48,6 +50,7 @@ class TestCANNversion(TestCase): | |||
| 48 | else: | 50 | else: |
| 49 | self.assertTrue(version == "", "When verssion_env < '25.', the result of get_cann_version is not right.") | 51 | self.assertTrue(version == "", "When verssion_env < '25.', the result of get_cann_version is not right.") |
| 50 | 52 | ||
| 53 | + | ||
| 51 | def test_compare_cann_version(self): | 54 | def test_compare_cann_version(self): |
| 52 | version_env = get_cann_version_from_env() | 55 | version_env = get_cann_version_from_env() |
| 53 | if not version_env.startswith("CANN") and version_env >= "8.1.RC1": | 56 | if not version_env.startswith("CANN") and version_env >= "8.1.RC1": |