Fix flaky test_export_cat.py lit test (upstream change).

llvm · Aug 23, 2021 · 177ccdd · 177ccdd
1 parent 32f56c6
commit 177ccdd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/frontends/pytorch/test/acap_export/test_export_cat.py b/frontends/pytorch/test/acap_export/test_export_cat.py
@@ -42,6 +42,6 @@ def forward(self, x):
 # 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)