Fix some ops' custom_fallback error.

DeepLink-org · Jan 18, 2024 · 9a0fdee · 9a0fdee
1 parent a802992
commit 9a0fdee
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/dicp/scripts/ci/ascend/test_env.sh b/dicp/scripts/ci/ascend/test_env.sh
@@ -4,3 +4,4 @@ LLAMA_MODEL_DIR=$1
 
 export DIPU_MOCK_CUDA=false
 export LLAMA_MODEL_DIR=$1
+export DIPU_KEEP_TORCHOP_DEFAULT_IMPL_OPS="rsqrt.out,_softmax.out"
diff --git a/dicp/scripts/ci/tops/ci_tops_test_env.sh b/dicp/scripts/ci/tops/ci_tops_test_env.sh
@@ -8,3 +8,4 @@ export DIPU_MOCK_CUDA=False
 # export LLAMA_MODEL_DIR=$1
 # export LLAMA_FINETUNE_DIR=$2
 # export STABLE_DIFFUSION_MODEL_DIR=$3
+export DIPU_KEEP_TORCHOP_DEFAULT_IMPL_OPS="rsqrt.out,_softmax.out"
Original file line number	Diff line number	Diff line change
Expand Up		@@ -4,3 +4,4 @@ LLAMA_MODEL_DIR=$1

		export DIPU_MOCK_CUDA=false
		export LLAMA_MODEL_DIR=$1
		export DIPU_KEEP_TORCHOP_DEFAULT_IMPL_OPS="rsqrt.out,_softmax.out"