From 51ec79310da258b8f7ed6591918f6e684a06e43c Mon Sep 17 00:00:00 2001 From: Lutz Roeder Date: Fri, 16 Dec 2022 11:57:28 -0500 Subject: [PATCH] Update backend test (#990) --- test/backend.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/backend.py b/test/backend.py index 44f02f52cd..e6c38afd46 100755 --- a/test/backend.py +++ b/test/backend.py @@ -45,13 +45,13 @@ def _test_torchscript(): # graph, _ = torch.jit._get_trace_graph(model, args) # pylint: disable=protected-access # torch.onnx._optimize_trace(graph, torch.onnx.OperatorExportTypes.ONNX) # trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'fasterrcnn_resnet50_fpn.pt')) + # torch.backends.quantized.engine = 'qnnpack' + # trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'd2go.pt')) # trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'mobilenetv2-quant_full-nnapi.pt')) - torch.backends.quantized.engine = 'qnnpack' - trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'd2go.pt')) # trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'inception_v3_traced.pt')) # trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'netron_issue_920.pt')) + # trace = torch.load(os.path.join(test_data_dir, 'pytorch', 'bert-base-uncased.pt')) torch._C._jit_pass_inline(trace.graph) - # https://github.com/pytorch/pytorch/blob/master/torch/csrc/jit/ir/ir.h netron.serve('resnet34', trace) # _test_onnx()