xref: /aosp_15_r20/external/llvm/test/CodeGen/NVPTX/simple-call.ll (revision 9880d6810fe72a1726cb53787c6711e909410d58) 
1*9880d681SAndroid Build Coastguard Worker; RUN: llc < %s -march=nvptx -mcpu=sm_20 | FileCheck %s
2*9880d681SAndroid Build Coastguard Worker; RUN: llc < %s -march=nvptx64 -mcpu=sm_20 | FileCheck %s
3*9880d681SAndroid Build Coastguard Worker
4*9880d681SAndroid Build Coastguard Worker
5*9880d681SAndroid Build Coastguard Worker
6*9880d681SAndroid Build Coastguard Worker; CHECK: .func ({{.*}}) device_func
7*9880d681SAndroid Build Coastguard Workerdefine float @device_func(float %a) noinline {
8*9880d681SAndroid Build Coastguard Worker  %ret = fmul float %a, %a
9*9880d681SAndroid Build Coastguard Worker  ret float %ret
10*9880d681SAndroid Build Coastguard Worker}
11*9880d681SAndroid Build Coastguard Worker
12*9880d681SAndroid Build Coastguard Worker; CHECK: .entry kernel_func
13*9880d681SAndroid Build Coastguard Workerdefine void @kernel_func(float* %a) {
14*9880d681SAndroid Build Coastguard Worker  %val = load float, float* %a
15*9880d681SAndroid Build Coastguard Worker; CHECK: call.uni (retval0),
16*9880d681SAndroid Build Coastguard Worker; CHECK: device_func,
17*9880d681SAndroid Build Coastguard Worker  %mul = call float @device_func(float %val)
18*9880d681SAndroid Build Coastguard Worker  store float %mul, float* %a
19*9880d681SAndroid Build Coastguard Worker  ret void
20*9880d681SAndroid Build Coastguard Worker}
21*9880d681SAndroid Build Coastguard Worker
22*9880d681SAndroid Build Coastguard Worker
23*9880d681SAndroid Build Coastguard Worker
24*9880d681SAndroid Build Coastguard Worker!nvvm.annotations = !{!1}
25*9880d681SAndroid Build Coastguard Worker
26*9880d681SAndroid Build Coastguard Worker!1 = !{void (float*)* @kernel_func, !"kernel", i32 1}
27