From 2affbb99bea9d7a8a95deaf2a90de20350193d36 Mon Sep 17 00:00:00 2001 From: zhanhao Date: Sat, 26 Oct 2024 18:37:04 +0800 Subject: [PATCH] add 8.0.T37,8.0.T38,8.0.T39 to foreach black list --- torch_npu/utils/_optim.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torch_npu/utils/_optim.py b/torch_npu/utils/_optim.py index 0eaeb63a164..78cb1d7b052 100644 --- a/torch_npu/utils/_optim.py +++ b/torch_npu/utils/_optim.py @@ -7,8 +7,8 @@ from torch_npu.utils.collect_env import get_cann_version _device_name = None _cann_version = get_cann_version() _foreach_black_list_for_cann_starts_with = ['8.0.RC1', '8.0.RC2'] -_foreach_black_list_for_cann_all = ['not known', '8.0.T1', '8.0.T2', '8.0.T3', '8.0.T37', '8.0.T5', '8.0.T6', '8.0.T7', - '8.0.T8', '8.0.T10', '8.0.T13', '8.0.T16', '8.0.T50', '8.0.T51', '8.0.T52'] +_foreach_black_list_for_cann_all = ['not known', '8.0.T1', '8.0.T2', '8.0.T3', '8.0.T5', '8.0.T6', '8.0.T7', + '8.0.T8', '8.0.T10', '8.0.T13', '8.0.T16', '8.0.T37', '8.0.T38', '8.0.T39', '8.0.T50', '8.0.T51', '8.0.T52'] def patch_supported_devices(): -- Gitee