llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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

pull/288/head
Stella Laurenzo 2021-08-22 19:54:01 -07:00 committed by Stella Laurenzo
parent 32f56c67f4
commit 177ccdd55b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frontends/pytorch/test/acap_export/test_export_cat.py
View File

@ -42,6 +42,6 @@ with mb.capture_function("conv_cat", [inputs, target]) as f:
# CHECK: "aten.convolution" # CHECK: "aten.convolution"
# CHECK: torch.prim.ListConstruct # CHECK: torch.prim.ListConstruct
# CHECK: "aten._cat" # CHECK: "aten._cat"
# CHECK: "aten._log_softmax" # CHECK: "aten._log_softmax.out"
# CHECK: "aten.nll_loss2d_forward" # CHECK: "aten.nll_loss2d_forward"
mb.module.operation.print(large_elements_limit=2) mb.module.operation.print(large_elements_limit=2)