已合并
[2.8.0]fix aoe ut #27468
daifu1234567创建于 2025年12月4日
[2.8.0]fix aoe ut #27468
已合并
共 1 个文件变更+2-0
| @@ -5,6 +5,7 @@ import torch_npu | |||
| 5 | 5 | ||
| 6 | from torch_npu.testing.testcase import TestCase, run_tests | 6 | from torch_npu.testing.testcase import TestCase, run_tests |
| 7 | from torch_npu.utils._path_manager import PathManager | 7 | from torch_npu.utils._path_manager import PathManager |
| 8 | +from torch_npu.testing.common_utils import SupportedDevices | ||
| 8 | 9 | ||
| 9 | 10 | ||
| 10 | class SmallModel(torch.nn.Module): | 11 | class SmallModel(torch.nn.Module): |
| @@ -21,6 +22,7 @@ class SmallModel(torch.nn.Module): | |||
| 21 | return input_1.reshape(input_1.shape[0], -1) | 22 | return input_1.reshape(input_1.shape[0], -1) |
| 22 | 23 | ||
| 23 | 24 | ||
| 25 | + | ||
| 24 | class TestAoe(TestCase): | 26 | class TestAoe(TestCase): |
| 25 | results_path = os.path.join(os.path.split(os.path.realpath(__file__))[0], "graphs") | 27 | results_path = os.path.join(os.path.split(os.path.realpath(__file__))[0], "graphs") |
| 26 | 28 | ||