[KUNLUNXIN] fix dipu ci

DeepLink-org · Jan 16, 2024 · 2f3ad7c · 2f3ad7c
1 parent c9c6120
commit 2f3ad7c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
@@ -561,6 +561,7 @@ jobs:
           source scripts/ci/ci_one_iter.sh export_pythonpath_kunlunxin ${basic_path}
           cd mmlab_pack
           rm -rf one_iter_data
+          XDNN_CONV_GEMM_DTYPE=int16 XPUAPI_DEBUG_CUSTOM_OP_LEVEL="broadcast_add,0x11" \
           python ../scripts/ci/ci_run_one_iter.py kunlunxin "" "" "" && rm -rf one_iter_data || (rm -rf one_iter_data && exit 1)
 
       - name: Perform cleanup one iter data