Skip to content

Commit d7ae03f

Browse files
committed
fixup tests
1 parent e092395 commit d7ae03f

File tree

2 files changed

+11
-5
lines changed

2 files changed

+11
-5
lines changed

llvm/test/CodeGen/NVPTX/bf16x2-instructions.ll

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -359,11 +359,12 @@ define <2 x bfloat> @test_select_cc_bf16_f32(<2 x bfloat> %a, <2 x bfloat> %b,
359359
define <2 x bfloat> @test_fptrunc_2xfloat(<2 x float> %a) #0 {
360360
; CHECK-LABEL: test_fptrunc_2xfloat(
361361
; CHECK: {
362-
; CHECK-NEXT: .reg .b64 %rd<2>;
362+
; CHECK-NEXT: .reg .b32 %r<4>;
363363
; CHECK-EMPTY:
364364
; CHECK-NEXT: // %bb.0:
365-
; CHECK-NEXT: ld.param.b64 %rd1, [test_fptrunc_2xfloat_param_0];
366-
; CHECK-NEXT: st.param.b32 [func_retval0], %rd1;
365+
; CHECK-NEXT: ld.param.v2.b32 {%r1, %r2}, [test_fptrunc_2xfloat_param_0];
366+
; CHECK-NEXT: cvt.rn.bf16x2.f32 %r3, %r2, %r1;
367+
; CHECK-NEXT: st.param.b32 [func_retval0], %r3;
367368
; CHECK-NEXT: ret;
368369
%r = fptrunc <2 x float> %a to <2 x bfloat>
369370
ret <2 x bfloat> %r

llvm/test/CodeGen/NVPTX/f16x2-instructions.ll

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1433,11 +1433,16 @@ define <2 x half> @test_sitofp_2xi32_fadd(<2 x i32> %a, <2 x half> %b) #0 {
14331433
define <2 x half> @test_fptrunc_2xfloat(<2 x float> %a) #0 {
14341434
; CHECK-LABEL: test_fptrunc_2xfloat(
14351435
; CHECK: {
1436+
; CHECK-NEXT: .reg .b16 %rs<3>;
1437+
; CHECK-NEXT: .reg .b32 %r<4>;
14361438
; CHECK-NEXT: .reg .b64 %rd<2>;
14371439
; CHECK-EMPTY:
14381440
; CHECK-NEXT: // %bb.0:
1439-
; CHECK-NEXT: ld.param.b64 %rd1, [test_fptrunc_2xfloat_param_0];
1440-
; CHECK-NEXT: st.param.b32 [func_retval0], %rd1;
1441+
; CHECK-NEXT: ld.param.v2.b32 {%r1, %r2}, [test_fptrunc_2xfloat_param_0];
1442+
; CHECK-NEXT: cvt.rn.f16.f32 %rs1, %r2;
1443+
; CHECK-NEXT: cvt.rn.f16.f32 %rs2, %r1;
1444+
; CHECK-NEXT: mov.b32 %r3, {%rs2, %rs1};
1445+
; CHECK-NEXT: st.param.b32 [func_retval0], %r3;
14411446
; CHECK-NEXT: ret;
14421447
%r = fptrunc <2 x float> %a to <2 x half>
14431448
ret <2 x half> %r

0 commit comments

Comments
 (0)