Skip to content

Commit

Permalink
Merge pull request #371 from DeepLink-org/wq/ci_one_iter4unet
Browse files Browse the repository at this point in the history
wq/add unet to test_one_iter_model
  • Loading branch information
caikun-pjlab authored Oct 30, 2023
2 parents 4256f13 + 5b7386c commit 90729be
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions dipu/scripts/ci/test_one_iter_model_list.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -96,5 +96,8 @@ cuda:
- model_cfg: "transformers examples/pytorch/question-answering/run_bert_qa.py workdirs_bert"

ascend:
# # mmsegmentation
- model_cfg: "mmsegmentation unet/unet-s5-d16_fcn_4xb4-160k_cityscapes-512x1024.py workdirs_unet"
fallback_op_list: "nll_loss2d_forward,nll_loss2d_backward,native_batch_norm,topk,convolution_overrideable,convolution_backward_overrideable,native_batch_norm_backward"
- 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"

0 comments on commit 90729be

Please sign in to comment.