Catch OSError when calling inspect.getfile (#113671)

Summary: Fixes #111328 X-link: pytorch/pytorch#113671 Approved by: https://github.com/Skylion007, https://github.com/williamwen42 Reviewed By: huydhn Differential Revision: D51336642 Pulled By: yanboliang fbshipit-source-id: 7f1aada2a7b3ab9611f35247394510d26889a35f
pytorch · Nov 15, 2023 · 10b1816 · 10b1816
1 parent a1c4fdf
commit 10b1816
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_dynamo/utils.py
@@ -739,7 +739,7 @@ def torchscript(model, example_inputs, verbose=False):
 def getfile(obj):
     try:
         return inspect.getfile(obj)
-    except TypeError:
+    except (TypeError, OSError):
         return None