Skip to content

Commit e374b13

Browse files
authored
set environment variable to turn on linear binary folding for SD&LCM (#2566)
1 parent 5824753 commit e374b13

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

models_v2/pytorch/LCM/inference/cpu/run_model.sh

+1
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ elif [ "${RUN_MODE}" == "compile-ipex" ]; then
9292
elif [[ "${RUN_MODE}" == "compile-inductor" || "1" == "${TORCH_INDUCTOR}" ]]; then
9393
export TORCHINDUCTOR_FREEZING=1
9494
export TORCHINDUCTOR_CPP_ENABLE_TILING_HEURISTIC=0
95+
export TORCHINDUCTOR_ENABLE_LINEAR_BINARY_FOLDING=1
9596
ARGS="$ARGS --compile_inductor"
9697
echo "### running torch.compile with inductor backend"
9798
else

models_v2/pytorch/stable_diffusion/inference/cpu/run_model.sh

+1
Original file line numberDiff line numberDiff line change
@@ -194,6 +194,7 @@ elif [ "${MODE}" == "compile-ipex" ]; then
194194
elif [[ "${MODE}" == "compile-inductor" || "1" == ${TORCH_INDUCTOR} ]]; then
195195
export TORCHINDUCTOR_FREEZING=1
196196
export TORCHINDUCTOR_CPP_ENABLE_TILING_HEURISTIC=0
197+
export TORCHINDUCTOR_ENABLE_LINEAR_BINARY_FOLDING=1
197198
ARGS="$ARGS --compile_inductor"
198199
echo "### running torch.compile with inductor backend"
199200
else

0 commit comments

Comments
 (0)