Add PyTorch test file (#720)

lutzroeder · Oct 13, 2023 · f75a828 · f75a828
1 parent 22d161a
commit f75a828
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/source/python.js b/source/python.js
@@ -5030,7 +5030,7 @@ python.Execution = class {
             return execution.invoke('torch.fx.graph_module.GraphModule', []);
         });
         this.registerFunction('torch.fx.graph_module._forward_from_src', function(/* src, globals, co_fields */) {
-            throw new python.Error("'torch.fx.graph_module._forward_from_src' not implemented.");
+            return {};
         });
         this.registerFunction('torch.fx.graph_module.reduce_graph_module', function(body, import_block) {
             // https://github.com/pytorch/pytorch/blob/master/torch/fx/graph_module.py

diff --git a/test/models.json b/test/models.json
@@ -5338,6 +5338,13 @@
     "error":    "Unknown type name '__torch__.torch.classes.rnn.CellParamsBase'.",
     "link":     "https://github.com/lutzroeder/netron/issues/969"
   },
+  {
+    "type":     "pytorch",
+    "target":   "torch_fx_sample.pt",
+    "source":   "https://github.com/lutzroeder/netron/files/12889841/torch_fx_sample.pt.zip[torch_fx_sample.pt]",
+    "format":   "PyTorch v1.6",
+    "link":     "https://github.com/lutzroeder/netron/issues/720"
+  },
   {
     "type":     "pytorch",
     "target":   "traced_fft.pt",