1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
; RUN: llc < %s -march=nvptx64 -mcpu=sm_30 -mattr=+ptx60 | FileCheck %s
declare i32 @llvm.nvvm.fns(i32, i32, i32)
; CHECK-LABEL: .func{{.*}}fns
define i32 @fns(i32 %mask, i32 %base, i32 %offset) {
; CHECK: ld.param.u32 [[MASK:%r[0-9]+]], [fns_param_0];
; CHECK: ld.param.u32 [[BASE:%r[0-9]+]], [fns_param_1];
; CHECK: ld.param.u32 [[OFFSET:%r[0-9]+]], [fns_param_2];
; CHECK: fns.b32 {{%r[0-9]+}}, [[MASK]], [[BASE]], [[OFFSET]];
%r0 = call i32 @llvm.nvvm.fns(i32 %mask, i32 %base, i32 %offset);
; CHECK: fns.b32 {{%r[0-9]+}}, [[MASK]], [[BASE]], 0;
%r1 = call i32 @llvm.nvvm.fns(i32 %mask, i32 %base, i32 0);
%r01 = add i32 %r0, %r1;
; CHECK: fns.b32 {{%r[0-9]+}}, [[MASK]], 1, [[OFFSET]];
%r2 = call i32 @llvm.nvvm.fns(i32 %mask, i32 1, i32 %offset);
; CHECK: fns.b32 {{%r[0-9]+}}, [[MASK]], 1, 0;
%r3 = call i32 @llvm.nvvm.fns(i32 %mask, i32 1, i32 0);
%r23 = add i32 %r2, %r3;
%r0123 = add i32 %r01, %r23;
; CHECK: fns.b32 {{%r[0-9]+}}, 2, [[BASE]], [[OFFSET]];
%r4 = call i32 @llvm.nvvm.fns(i32 2, i32 %base, i32 %offset);
; CHECK: fns.b32 {{%r[0-9]+}}, 2, [[BASE]], 0;
%r5 = call i32 @llvm.nvvm.fns(i32 2, i32 %base, i32 0);
%r45 = add i32 %r4, %r5;
; CHECK: fns.b32 {{%r[0-9]+}}, 2, 1, [[OFFSET]];
%r6 = call i32 @llvm.nvvm.fns(i32 2, i32 1, i32 %offset);
; CHECK: fns.b32 {{%r[0-9]+}}, 2, 1, 0;
%r7 = call i32 @llvm.nvvm.fns(i32 2, i32 1, i32 0);
%r67 = add i32 %r6, %r7;
%r4567 = add i32 %r45, %r67;
%r = add i32 %r0123, %r4567;
ret i32 %r;
}
|
