xref: /aosp_15_r20/external/llvm/test/CodeGen/NVPTX/vector-args.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58)
1*9880d681SAndroid Build Coastguard Worker; RUN: llc < %s -march=nvptx -mcpu=sm_20 | FileCheck %s
2*9880d681SAndroid Build Coastguard Worker
3*9880d681SAndroid Build Coastguard Worker
4*9880d681SAndroid Build Coastguard Workerdefine float @foo(<2 x float> %a) {
5*9880d681SAndroid Build Coastguard Worker; CHECK: .func (.param .b32 func_retval0) foo
6*9880d681SAndroid Build Coastguard Worker; CHECK: .param .align 8 .b8 foo_param_0[8]
7*9880d681SAndroid Build Coastguard Worker; CHECK: ld.param.v2.f32 {%f{{[0-9]+}}, %f{{[0-9]+}}}
8*9880d681SAndroid Build Coastguard Worker  %t1 = fmul <2 x float> %a, %a
9*9880d681SAndroid Build Coastguard Worker  %t2 = extractelement <2 x float> %t1, i32 0
10*9880d681SAndroid Build Coastguard Worker  %t3 = extractelement <2 x float> %t1, i32 1
11*9880d681SAndroid Build Coastguard Worker  %t4 = fadd float %t2, %t3
12*9880d681SAndroid Build Coastguard Worker  ret float %t4
13*9880d681SAndroid Build Coastguard Worker}
14*9880d681SAndroid Build Coastguard Worker
15*9880d681SAndroid Build Coastguard Worker
16*9880d681SAndroid Build Coastguard Workerdefine float @bar(<4 x float> %a) {
17*9880d681SAndroid Build Coastguard Worker; CHECK: .func (.param .b32 func_retval0) bar
18*9880d681SAndroid Build Coastguard Worker; CHECK: .param .align 16 .b8 bar_param_0[16]
19*9880d681SAndroid Build Coastguard Worker; CHECK: ld.param.v4.f32 {%f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}}}
20*9880d681SAndroid Build Coastguard Worker  %t1 = fmul <4 x float> %a, %a
21*9880d681SAndroid Build Coastguard Worker  %t2 = extractelement <4 x float> %t1, i32 0
22*9880d681SAndroid Build Coastguard Worker  %t3 = extractelement <4 x float> %t1, i32 1
23*9880d681SAndroid Build Coastguard Worker  %t4 = fadd float %t2, %t3
24*9880d681SAndroid Build Coastguard Worker  ret float %t4
25*9880d681SAndroid Build Coastguard Worker}
26*9880d681SAndroid Build Coastguard Worker
27*9880d681SAndroid Build Coastguard Worker
28*9880d681SAndroid Build Coastguard Workerdefine <4 x float> @baz(<4 x float> %a) {
29*9880d681SAndroid Build Coastguard Worker; CHECK: .func  (.param .align 16 .b8 func_retval0[16]) baz
30*9880d681SAndroid Build Coastguard Worker; CHECK: .param .align 16 .b8 baz_param_0[16]
31*9880d681SAndroid Build Coastguard Worker; CHECK: ld.param.v4.f32 {%f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}}}
32*9880d681SAndroid Build Coastguard Worker; CHECK: st.param.v4.f32 [func_retval0+0], {%f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}}, %f{{[0-9]+}}}
33*9880d681SAndroid Build Coastguard Worker  %t1 = fmul <4 x float> %a, %a
34*9880d681SAndroid Build Coastguard Worker  ret <4 x float> %t1
35*9880d681SAndroid Build Coastguard Worker}
36