Skip to content

Commit d7af7ba

Browse files
committed
use max-autotune
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent e3b970c commit d7af7ba

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/ipex/modeling_base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -242,7 +242,7 @@ def apply_torch_compile(self):
242242
inductor_config.cpp_wrapper = True
243243
os.environ["TORCHINDUCTOR_FREEZING"] = "1"
244244
logger.info("Enable torch.compile optimization")
245-
self.model.forward = torch.compile(self.model.forward)
245+
self.model.forward = torch.compile(self.model.forward, mode="max-autotune")
246246

247247
def _init_warmup(self):
248248
inputs = prepare_jit_inputs(self.model, self.export_feature, False)

0 commit comments

Comments
 (0)