@@ -224,7 +224,7 @@ define i32 @cvt_rna_tf32_f32(float %f1) {
224
224
declare i32 @llvm.nvvm.f2tf32.rna (float )
225
225
226
226
227
- define <2 x bfloat> @fold_ff2bf16x2 (float %a , float %b ) {
227
+ define <2 x bfloat> @fold_ff2bf16x2 (float %lo , float %hi ) {
228
228
; CHECK-LABEL: fold_ff2bf16x2(
229
229
; CHECK: {
230
230
; CHECK-NEXT: .reg .b32 %r<2>;
@@ -236,14 +236,14 @@ define <2 x bfloat> @fold_ff2bf16x2(float %a, float %b) {
236
236
; CHECK-NEXT: cvt.rn.bf16x2.f32 %r1, %f2, %f1;
237
237
; CHECK-NEXT: st.param.b32 [func_retval0], %r1;
238
238
; CHECK-NEXT: ret;
239
- %ah = fptrunc float %a to bfloat
240
- %bh = fptrunc float %b to bfloat
241
- %v0 = insertelement <2 x bfloat> poison, bfloat %ah , i64 0
242
- %v1 = insertelement <2 x bfloat> %v0 , bfloat %bh , i64 1
239
+ %loh = fptrunc float %lo to bfloat
240
+ %hih = fptrunc float %hi to bfloat
241
+ %v0 = insertelement <2 x bfloat> poison, bfloat %loh , i64 0
242
+ %v1 = insertelement <2 x bfloat> %v0 , bfloat %hih , i64 1
243
243
ret <2 x bfloat> %v1
244
244
}
245
245
246
- define <2 x half > @fold_ff2f16x2 (float %a , float %b ) {
246
+ define <2 x half > @fold_ff2f16x2 (float %lo , float %hi ) {
247
247
; CHECK-LABEL: fold_ff2f16x2(
248
248
; CHECK: {
249
249
; CHECK-NEXT: .reg .b32 %r<2>;
@@ -255,9 +255,9 @@ define <2 x half> @fold_ff2f16x2(float %a, float %b) {
255
255
; CHECK-NEXT: cvt.rn.f16x2.f32 %r1, %f2, %f1;
256
256
; CHECK-NEXT: st.param.b32 [func_retval0], %r1;
257
257
; CHECK-NEXT: ret;
258
- %ah = fptrunc float %a to half
259
- %bh = fptrunc float %b to half
260
- %v0 = insertelement <2 x half > poison, half %ah , i64 0
261
- %v1 = insertelement <2 x half > %v0 , half %bh , i64 1
258
+ %loh = fptrunc float %lo to half
259
+ %hih = fptrunc float %hi to half
260
+ %v0 = insertelement <2 x half > poison, half %loh , i64 0
261
+ %v1 = insertelement <2 x half > %v0 , half %hih , i64 1
262
262
ret <2 x half > %v1
263
263
}
0 commit comments