Rebase on main

openxla · Jan 19, 2023 · 591dcb2 · 591dcb2
1 parent e9e5153
commit 591dcb2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/stablehlo/tests/legalize_stablehlo_to_vhlo.mlir b/stablehlo/tests/legalize_stablehlo_to_vhlo.mlir
@@ -730,7 +730,7 @@ func.func @op_dot(%arg0: tensor<8x16xf32>, %arg1: tensor<16x8xf32>) -> tensor<8x
     precision_config = []
   } : (tensor<8x16xf32>, tensor<16x8xf32>) -> tensor<8x8xf32>
   //      CHECK: "vhlo.dot"(%arg0, %arg1)
-  // CHECK-SAME: : (tensor<8x16xf32>, tensor<16x8xf32>) -> tensor<8x8xf32>
+  // CHECK-SAME: : (!vhlo.tensor<8x16x!vhlo.f32>, !vhlo.tensor<16x8x!vhlo.f32>) -> !vhlo.tensor<8x8x!vhlo.f32>
   %1 = "stablehlo.dot"(%arg0, %arg1) : (tensor<8x16xf32>, tensor<16x8xf32>) -> tensor<8x8xf32>
   func.return %0 : tensor<8x8xf32>
 }