We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bbf09cd commit 581aa6fCopy full SHA for 581aa6f
optimum/intel/ipex/modeling_base.py
@@ -253,7 +253,7 @@ def apply_torch_compile(self):
253
inductor_config.cpp_wrapper = True
254
os.environ["TORCHINDUCTOR_FREEZING"] = "1"
255
logger.info("Enable torch.compile optimization")
256
- self.model.forward = torch.compile(self.model.forward, mode="max-autotune")
+ self.model.forward = torch.compile(self.model.forward)
257
258
def _init_warmup(self):
259
inputs = prepare_jit_inputs(self.model, self.export_feature, False)
0 commit comments