Skip to content

Commit

Permalink
remove _softmax.out custom fallback
Browse files Browse the repository at this point in the history
  • Loading branch information
@tangzhiyi11
tangzhiyi11 committed Jan 17, 2024
1 parent 452ad27 commit f510221
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dicp/scripts/ci/ascend/dipu_env.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,4 +2,4 @@

export DIPU_DEVICE=ascend
export DIPU_WITH_DIOPI_LIBRARY=DISABLE
export DIPU_KEEP_TORCHOP_DEFAULT_IMPL_OPS="rsqrt.out,mm,linear"
export DIPU_KEEP_TORCHOP_DEFAULT_IMPL_OPS="rsqrt.out,mm,linear,_softmax.out"

0 comments on commit f510221

Please sign in to comment.