| ; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 6 |
| ; RUN: llc < %s -mtriple=nvptx64 -mcpu=sm_90 -mattr=+ptx81| FileCheck --check-prefixes=CHECK %s |
| ; RUN: %if ptxas-sm_90 && ptxas-isa-8.1 %{ llc < %s -mtriple=nvptx64 -mcpu=sm_90 -mattr=+ptx81| %ptxas-verify -arch=sm_90 %} |
| |
| define i32 @test_aggr_smem_size() { |
| ; CHECK-LABEL: test_aggr_smem_size( |
| ; CHECK: { |
| ; CHECK-NEXT: .reg .b32 %r<2>; |
| ; CHECK-EMPTY: |
| ; CHECK-NEXT: // %bb.0: |
| ; CHECK-NEXT: mov.u32 %r1, %aggr_smem_size; |
| ; CHECK-NEXT: st.param.b32 [func_retval0], %r1; |
| ; CHECK-NEXT: ret; |
| %a = tail call i32 @llvm.nvvm.read.ptx.sreg.aggr_smem_size() |
| ret i32 %a |
| } |
| |
| declare i32 @llvm.nvvm.read.ptx.sreg.aggr_smem_size() |
