From 91e075c6193c554373475ae253b2b035084f5a1f Mon Sep 17 00:00:00 2001 From: AmosLewis Date: Fri, 17 Jan 2025 13:45:02 -0800 Subject: [PATCH] Structure the code same as elementwise$unary --- test/Conversion/TorchToLinalg/elementwise.mlir | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/test/Conversion/TorchToLinalg/elementwise.mlir b/test/Conversion/TorchToLinalg/elementwise.mlir index d126f700c9a2c..f9a6ff5c1aacb 100644 --- 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