When transposing the last two axes of a shape with 1, 1 as its last two strides, gemm does not correctly deduce that the input with that shape is transposed, due to it only checking the strides. This causes the rocBLAS implementation of gemm to fail. The issue doesn't appear when using MLIR.
When transposing the last two axes of a shape with
1, 1
as its last two strides, gemm does not correctly deduce that the input with that shape is transposed, due to it only checking the strides. This causes the rocBLAS implementation of gemm to fail. The issue doesn't appear when using MLIR.