torch-mlir icon indicating copy to clipboard operation
torch-mlir copied to clipboard

Published 20 hours ago • verified publisher icon llvm

Add some unit tests to check i32 shape

Open Yancey1989 opened this issue 2 years ago • 3 comments

In the current codebase, we can build mhlo with TORCH_MLIR_ENABLE_MHLO_TRUNC_DIMSIZE_TO_I32 to enable the i32 shape, but there are no unit tests to test the i32 feature, we need a way to test the i32 shape.

Yancey1989 avatar Aug 09 '22 06:08 Yancey1989

This will be easy to fix after https://github.com/llvm/torch-mlir/issues/1227

silvasean avatar Aug 15 '22 23:08 silvasean

@Yancey1989 it looks like https://github.com/llvm/torch-mlir/issues/1227 landed -- could you try to add these tests now?

silvasean avatar Oct 07 '22 13:10 silvasean

That’s great, I will try it asap.

Yancey1989 avatar Oct 10 '22 03:10 Yancey1989