Skip to content

Commit 1b7922f

Browse files
authored
modify fallback path of graph capture (#1482) (#1485)
1 parent aa88ba6 commit 1b7922f

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

intel_extension_for_pytorch/frontend.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,7 @@ def forward(*input, **kwargs):
227227
except:
228228
warnings.warn("Both JIT and TorchDynamo failed, fallback to original model.")
229229
self.method = RunMethods.EagerInfer
230+
torch._dynamo.reset()
230231
return self.model(*input, **kwargs)
231232

232233
return forward

0 commit comments

Comments
 (0)