From 177ccdd55beecbf20b4f8796c2f615c905f36ae3 Mon Sep 17 00:00:00 2001 From: Stella Laurenzo Date: Sun, 22 Aug 2021 19:54:01 -0700 Subject: [PATCH] Fix flaky test_export_cat.py lit test (upstream change). --- frontends/pytorch/test/acap_export/test_export_cat.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/frontends/pytorch/test/acap_export/test_export_cat.py b/frontends/pytorch/test/acap_export/test_export_cat.py index f26026413..723e4fa88 100644 --- a/frontends/pytorch/test/acap_export/test_export_cat.py +++ b/frontends/pytorch/test/acap_export/test_export_cat.py @@ -42,6 +42,6 @@ with mb.capture_function("conv_cat", [inputs, target]) as f: # CHECK: "aten.convolution" # CHECK: torch.prim.ListConstruct # CHECK: "aten._cat" -# CHECK: "aten._log_softmax" +# CHECK: "aten._log_softmax.out" # CHECK: "aten.nll_loss2d_forward" mb.module.operation.print(large_elements_limit=2)