Structure the code same as elementwise$unary

llvm · Jan 17, 2025 · 91e075c · 91e075c
1 parent 3a1153a
commit 91e075c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/Conversion/TorchToLinalg/elementwise.mlir b/test/Conversion/TorchToLinalg/elementwise.mlir
@@ -105,9 +105,9 @@ func.func @elementwise_sinh(%arg0: !torch.vtensor<[3],f32>) -> !torch.vtensor<[3
 
 // -----
 
-// CHECK-LABEL:   func.func @elementwise_todtype_bf162f16(
-// CHECK-SAME:        %[[VAL_0:.*]]: !torch.vtensor<[1,?,32,128],bf16>) -> !torch.vtensor<[1,?,32,128],f16> {
-// CHECK:           %[[INPUT:.*]] = torch_c.to_builtin_tensor %[[VAL_0]] : !torch.vtensor<[1,?,32,128],bf16> -> tensor<1x?x32x128xbf16>
+// CHECK-LABEL:   func.func @elementwise$todtype(
+// CHECK-SAME:                            %[[ARG:.*]]: !torch.vtensor<[1,?,32,128],bf16>) -> !torch.vtensor<[1,?,32,128],f16> {
+// CHECK:           %[[INPUT:.*]] = torch_c.to_builtin_tensor %[[ARG]] : !torch.vtensor<[1,?,32,128],bf16> -> tensor<1x?x32x128xbf16>
 // CHECK:           %[[INT5:.*]] = torch.constant.int 5
 // CHECK:           %[[FALSE:.*]] = torch.constant.bool false
 // CHECK:           %[[NONE:.*]] = torch.constant.none