diff --git a/dipu/scripts/ci/test_one_iter_traditional_model_list.yaml b/dipu/scripts/ci/test_one_iter_traditional_model_list.yaml index 54585ffcc..12542f7e2 100644 --- a/dipu/scripts/ci/test_one_iter_traditional_model_list.yaml +++ b/dipu/scripts/ci/test_one_iter_traditional_model_list.yaml @@ -103,7 +103,12 @@ ascend: # - model_cfg: "mmdetection detr/detr_r50_8xb2-150e_coco.py workdirs_detr" # fallback_op_list: "fill_.Scalar,baddbmm.out,where.self,linear_backward,linear,uniform_,any.all_out,_foreach_addcdiv_.ScalarList,native_batch_norm_backward,convolution_overrideable" # mmpretrain - # - model_cfg: "mmpretrain resnet/resnet50_8xb32_in1k.py workdirs_resnet" + - model_cfg: "mmpretrain convnext/convnext-small_32xb128_in1k.py workdirs_convnext --no-pin-memory" + fallback_op_list: "convolution_backward_overrideable,convolution_overrideable" + - model_cfg: "mmpretrain efficientnet/efficientnet-b2_8xb32_in1k.py workdirs_efficientnet --no-pin-memory" + fallback_op_list: "convolution_backward_overrideable,convolution_overrideable" + - model_cfg: "mmpretrain resnet/resnet50_8xb32_in1k.py workdirs_resnet --no-pin-memory" + fallback_op_list: "convolution_backward_overrideable,convolution_overrideable" kunlunxin: # mmpretrain