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

Fix CI failure due to upstream PyTorch change in aten.mean.dim op 

Fixes https://github.com/llvm/torch-mlir/issues/1121

Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
pull/1113/head
Vivek Khandelwal 2022-07-29 12:57:07 +05:30
parent b389053653
commit 9a1203c844
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/torch_mlir_e2e_test/test_suite/stats.py
View File

@ -97,7 +97,7 @@ class MeanDimDtypeModule(torch.nn.Module):
([-1, -1, -1], torch.float64, True), ([-1, -1, -1], torch.float64, True),
]) ])
def forward(self, x): def forward(self, x):
return torch.ops.aten.mean(x, 0, dtype=torch.float32) return torch.ops.aten.mean(x, (0,), dtype=torch.float32)
@register_test_case(module_factory=lambda: MeanDimDtypeModule()) @register_test_case(module_factory=lambda: MeanDimDtypeModule())