已合并
Skip test_grad_scaling_unscale. #19223
xiaxia3创建于 2025年3月20日
Skip test_grad_scaling_unscale. #19223
已合并
从refs/pull/19223/head合入到master
共 1 个文件变更+3-1
| @@ -67,4 +67,6 @@ class TestAmpForeachNonFiniteCheckAndUnscale(TestCase): | |||
| 67 | 67 | ||
| 68 | 68 | ||
| 69 | if __name__ == "__main__": | 69 | if __name__ == "__main__": |
| 70 | - run_tests() | 70 | + device_name = torch_npu.npu.get_device_name(0)[:10] |
| 71 | + if device_name in ["Ascend910A"]: | ||
| 72 | + run_tests() | ||