Fix test

Author: dhernandez0

mlir/test/rocmlir-gen/gemm-elementwise-gemm-kernel.mlir

@@ -19,6 +19,6 @@
 // CHECK: return
 
 // CHECK-LABEL: func.func @host_naive_gemm_gemm
-// CHECK: %[[abTensor:.*]] = tosa.matmul %[[aTensor:.*]], %[[bTensor:.*]] : ([[aShape:tensor<.*>]], [[bShape:tensor<.*>]]) -> [[squareShape:tensor<.*>]]
-// CHECK-DAG: %[[resultTensor:.*]] = tosa.matmul %[[abTensor]], %[[cTensor:.*]] : ([[squareShape]], [[cShape:tensor<.*>]]) -> [[cShape]]
+// CHECK: %[[abTensor:.*]] = tosa.matmul %[[aTensor:.*]], %[[bTensor:.*]], %{{.*}}, %{{.*}} : ([[aShape:tensor<.*>]], [[bShape:tensor<.*>]], tensor<1xf32>, tensor<1xf32>) -> [[squareShape:tensor<.*>]]
+// CHECK-DAG: %[[resultTensor:.*]] = tosa.matmul %[[abTensor]], %[[cTensor:.*]], %{{.*}}, %{{.*}} : ([[squareShape]], [[cShape:tensor<.*>]], tensor<1xf32>, tensor<1xf32>) -> [[cShape]]
 // CHECK: return