36 lines · plain
1; NOTE: Assertions have been autogenerated by utils/update_llc_test_checks.py UTC_ARGS: --version 62; RUN: llc < %s -mtriple=nvptx64 -mcpu=sm_80 -mattr=+ptx70| FileCheck --check-prefixes=CHECK-PTX64 %s3; RUN: llc < %s -mtriple=nvptx64 -mcpu=sm_80 -mattr=+ptx70 --nvptx-short-ptr| FileCheck --check-prefixes=CHECK-PTX-SHARED32 %s4; RUN: %if ptxas-sm_80 && ptxas-isa-7.0 %{ llc < %s -mtriple=nvptx64 -mcpu=sm_80 -mattr=+ptx70| %ptxas-verify -arch=sm_80 %}5; RUN: %if ptxas-sm_80 && ptxas-isa-7.0 %{ llc < %s -mtriple=nvptx64 -mcpu=sm_80 -mattr=+ptx70 --nvptx-short-ptr| %ptxas-verify -arch=sm_80 %}6 7declare i1 @llvm.nvvm.mbarrier.test.wait.scope.cta.space.cta(ptr addrspace(3), i64)8 9define void @mbar_test_wait(ptr addrspace(3) %mbar, i64 %state) {10; CHECK-PTX64-LABEL: mbar_test_wait(11; CHECK-PTX64: {12; CHECK-PTX64-NEXT: .reg .pred %p<2>;13; CHECK-PTX64-NEXT: .reg .b64 %rd<3>;14; CHECK-PTX64-EMPTY:15; CHECK-PTX64-NEXT: // %bb.0:16; CHECK-PTX64-NEXT: ld.param.b64 %rd1, [mbar_test_wait_param_0];17; CHECK-PTX64-NEXT: ld.param.b64 %rd2, [mbar_test_wait_param_1];18; CHECK-PTX64-NEXT: mbarrier.test_wait.shared.b64 %p1, [%rd1], %rd2;19; CHECK-PTX64-NEXT: ret;20;21; CHECK-PTX-SHARED32-LABEL: mbar_test_wait(22; CHECK-PTX-SHARED32: {23; CHECK-PTX-SHARED32-NEXT: .reg .pred %p<2>;24; CHECK-PTX-SHARED32-NEXT: .reg .b32 %r<2>;25; CHECK-PTX-SHARED32-NEXT: .reg .b64 %rd<2>;26; CHECK-PTX-SHARED32-EMPTY:27; CHECK-PTX-SHARED32-NEXT: // %bb.0:28; CHECK-PTX-SHARED32-NEXT: ld.param.b32 %r1, [mbar_test_wait_param_0];29; CHECK-PTX-SHARED32-NEXT: ld.param.b64 %rd1, [mbar_test_wait_param_1];30; CHECK-PTX-SHARED32-NEXT: mbarrier.test_wait.shared.b64 %p1, [%r1], %rd1;31; CHECK-PTX-SHARED32-NEXT: ret;32 %ret0 = call i1 @llvm.nvvm.mbarrier.test.wait.scope.cta.space.cta(ptr addrspace(3) %mbar, i64 %state)33 34 ret void35}36