xref: /llvm-project/llvm/test/CodeGen/NVPTX/dynamic_stackalloc.ll (revision 0068078dca60b41ad1c7bdd4448e7de718b82a5d)
1; RUN: not llc < %s -mtriple=nvptx -mattr=+ptx72 -mcpu=sm_52 2>&1 | FileCheck %s --check-prefixes=CHECK-FAILS
2; RUN: not llc < %s -mtriple=nvptx -mattr=+ptx73 -mcpu=sm_50 2>&1 | FileCheck %s --check-prefixes=CHECK-FAILS
3
4; RUN: llc < %s -mtriple=nvptx -mattr=+ptx73 -mcpu=sm_52 | FileCheck %s --check-prefixes=CHECK,CHECK-32
5; RUN: llc < %s -mtriple=nvptx64 -mattr=+ptx73 -mcpu=sm_52 | FileCheck %s --check-prefixes=CHECK,CHECK-64
6; RUN: %if ptxas && !ptxas-12.0 %{ llc < %s -mtriple=nvptx -mattr=+ptx73 -mcpu=sm_52 | %ptxas-verify %}
7; RUN: %if ptxas %{ llc < %s -mtriple=nvptx64 -mattr=+ptx73 -mcpu=sm_52 | %ptxas-verify %}
8
9; CHECK-FAILS: in function test_dynamic_stackalloc{{.*}}: Support for dynamic alloca introduced in PTX ISA version 7.3 and requires target sm_52.
10
11; CHECK-LABEL: .visible .func  (.param .b32 func_retval0) test_dynamic_stackalloc(
12; CHECK-NOT: __local_depot
13
14; CHECK-32:       ld.param.u32  %r[[SIZE:[0-9]]], [test_dynamic_stackalloc_param_0];
15; CHECK-32-NEXT:  add.s32 %r[[SIZE2:[0-9]]], %r[[SIZE]], 7;
16; CHECK-32-NEXT:  and.b32         %r[[SIZE3:[0-9]]], %r[[SIZE2]], -8;
17; CHECK-32-NEXT:  alloca.u32  %r[[ALLOCA:[0-9]]], %r[[SIZE3]], 16;
18; CHECK-32-NEXT:  cvta.local.u32  %r[[ALLOCA]], %r[[ALLOCA]];
19; CHECK-32-NEXT:  { // callseq 0, 0
20; CHECK-32-NEXT:  .param .b32 param0;
21; CHECK-32-NEXT:  st.param.b32  [param0], %r[[ALLOCA]];
22
23; CHECK-64:       ld.param.u64  %rd[[SIZE:[0-9]]], [test_dynamic_stackalloc_param_0];
24; CHECK-64-NEXT:  add.s64 %rd[[SIZE2:[0-9]]], %rd[[SIZE]], 7;
25; CHECK-64-NEXT:  and.b64 %rd[[SIZE3:[0-9]]], %rd[[SIZE2]], -8;
26; CHECK-64-NEXT:  alloca.u64  %rd[[ALLOCA:[0-9]]], %rd[[SIZE3]], 16;
27; CHECK-64-NEXT:  cvta.local.u64  %rd[[ALLOCA]], %rd[[ALLOCA]];
28; CHECK-64-NEXT:  { // callseq 0, 0
29; CHECK-64-NEXT:  .param .b64 param0;
30; CHECK-64-NEXT:  st.param.b64  [param0], %rd[[ALLOCA]];
31
32; CHECK-NEXT:     .param .b32 retval0;
33; CHECK-NEXT:     call.uni (retval0),
34; CHECK-NEXT:     bar,
35
36define i32 @test_dynamic_stackalloc(i64 %n) {
37  %alloca = alloca i8, i64 %n, align 16
38  %call = call i32 @bar(ptr %alloca)
39  ret i32 %call
40}
41
42declare i32 @bar(ptr)
43