Update backend test (#990)

lutzroeder · Apr 13, 2023 · 9611986 · 9611986
1 parent 8de0000
commit 9611986
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/test/backend.py b/test/backend.py
@@ -86,6 +86,13 @@ def _test_torchscript_nnapi():
     torch._C._jit_pass_inline(trace.graph) # pylint: disable=protected-access
     netron.serve('inception_v3', trace)
 
+def _test_torchscript_alexnet():
+    torch = __import__('torch')
+    trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'alexnet.pt'))
+    torch._C._jit_pass_inline(trace.graph) # pylint: disable=protected-access
+    netron.serve('alexnet', trace)
+
+
 # _test_onnx()
 # _test_onnx_iterate()
 
@@ -96,4 +103,5 @@ def _test_torchscript_nnapi():
 # _test_torchscript_scalar()
 # _test_torchscript_tuple()
 # _test_torchscript_nnapi()
-_test_torchscript_transformer()
+# _test_torchscript_transformer()
+_test_torchscript_alexnet()