File tree 2 files changed +2
-0
lines changed
stable_diffusion/inference/cpu
2 files changed +2
-0
lines changed Original file line number Diff line number Diff line change @@ -92,6 +92,7 @@ elif [ "${RUN_MODE}" == "compile-ipex" ]; then
92
92
elif [[ " ${RUN_MODE} " == " compile-inductor" || " 1" == " ${TORCH_INDUCTOR} " ]]; then
93
93
export TORCHINDUCTOR_FREEZING=1
94
94
export TORCHINDUCTOR_CPP_ENABLE_TILING_HEURISTIC=0
95
+ export TORCHINDUCTOR_ENABLE_LINEAR_BINARY_FOLDING=1
95
96
ARGS=" $ARGS --compile_inductor"
96
97
echo " ### running torch.compile with inductor backend"
97
98
else
Original file line number Diff line number Diff line change @@ -194,6 +194,7 @@ elif [ "${MODE}" == "compile-ipex" ]; then
194
194
elif [[ " ${MODE} " == " compile-inductor" || " 1" == ${TORCH_INDUCTOR} ]]; then
195
195
export TORCHINDUCTOR_FREEZING=1
196
196
export TORCHINDUCTOR_CPP_ENABLE_TILING_HEURISTIC=0
197
+ export TORCHINDUCTOR_ENABLE_LINEAR_BINARY_FOLDING=1
197
198
ARGS=" $ARGS --compile_inductor"
198
199
echo " ### running torch.compile with inductor backend"
199
200
else
You can’t perform that action at this time.
0 commit comments