From 961198655b00b7d7323bad8c7d954ba6c1c868f4 Mon Sep 17 00:00:00 2001 From: Lutz Roeder Date: Thu, 13 Apr 2023 09:07:05 -0400 Subject: [PATCH] Update backend test (#990) --- test/backend.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/test/backend.py b/test/backend.py index a27bed8178..8ab30b0893 100755 --- 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()