llvmbot wrote:
<!--LLVM PR SUMMARY COMMENT--> @llvm/pr-subscribers-llvm-transforms Author: Matt Arsenault (arsenm) <details> <summary>Changes</summary> --- Full diff: https://github.com/llvm/llvm-project/pull/102010.diff 2 Files Affected: - (modified) llvm/lib/Transforms/Scalar/InferAddressSpaces.cpp (+3-1) - (added) llvm/test/Transforms/InferAddressSpaces/AMDGPU/is.constant.ll (+35) ``````````diff diff --git a/llvm/lib/Transforms/Scalar/InferAddressSpaces.cpp b/llvm/lib/Transforms/Scalar/InferAddressSpaces.cpp index 2ddf24be67702..67d4e7b5ee146 100644 --- a/llvm/lib/Transforms/Scalar/InferAddressSpaces.cpp +++ b/llvm/lib/Transforms/Scalar/InferAddressSpaces.cpp @@ -401,7 +401,8 @@ bool InferAddressSpacesImpl::rewriteIntrinsicOperands(IntrinsicInst *II, II->setCalledFunction(NewDecl); return true; } - case Intrinsic::prefetch: { + case Intrinsic::prefetch: + case Intrinsic::is_constant: { Function *NewDecl = Intrinsic::getDeclaration(M, II->getIntrinsicID(), {NewV->getType()}); II->setArgOperand(0, NewV); @@ -426,6 +427,7 @@ void InferAddressSpacesImpl::collectRewritableIntrinsicOperands( switch (IID) { case Intrinsic::ptrmask: case Intrinsic::objectsize: + case Intrinsic::is_constant: appendsFlatAddressExpressionToPostorderStack(II->getArgOperand(0), PostorderStack, Visited); break; diff --git a/llvm/test/Transforms/InferAddressSpaces/AMDGPU/is.constant.ll b/llvm/test/Transforms/InferAddressSpaces/AMDGPU/is.constant.ll new file mode 100644 index 0000000000000..767681a281792 --- /dev/null +++ b/llvm/test/Transforms/InferAddressSpaces/AMDGPU/is.constant.ll @@ -0,0 +1,35 @@ +; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 5 +; RUN: opt -S -mtriple=amdgcn-amd-amdhsa -passes=infer-address-spaces %s | FileCheck %s + +define i1 @is_constant_global_to_flat(ptr addrspace(1) %ptr) { +; CHECK-LABEL: define i1 @is_constant_global_to_flat( +; CHECK-SAME: ptr addrspace(1) [[PTR:%.*]]) { +; CHECK-NEXT: [[RET:%.*]] = call i1 @llvm.is.constant.p1(ptr addrspace(1) [[PTR]]) +; CHECK-NEXT: ret i1 [[RET]] +; + %cast = addrspacecast ptr addrspace(1) %ptr to ptr + %ret = call i1 @llvm.is.constant.p0(ptr %cast) + ret i1 %ret +} + +define i1 @is_constant_local_to_flat(ptr addrspace(3) %ptr) { +; CHECK-LABEL: define i1 @is_constant_local_to_flat( +; CHECK-SAME: ptr addrspace(3) [[PTR:%.*]]) { +; CHECK-NEXT: [[RET:%.*]] = call i1 @llvm.is.constant.p3(ptr addrspace(3) [[PTR]]) +; CHECK-NEXT: ret i1 [[RET]] +; + %cast = addrspacecast ptr addrspace(3) %ptr to ptr + %ret = call i1 @llvm.is.constant.p0(ptr %cast) + ret i1 %ret +} + +define i1 @is_constant_private_to_flat(ptr addrspace(5) %ptr) { +; CHECK-LABEL: define i1 @is_constant_private_to_flat( +; CHECK-SAME: ptr addrspace(5) [[PTR:%.*]]) { +; CHECK-NEXT: [[RET:%.*]] = call i1 @llvm.is.constant.p5(ptr addrspace(5) [[PTR]]) +; CHECK-NEXT: ret i1 [[RET]] +; + %cast = addrspacecast ptr addrspace(5) %ptr to ptr + %ret = call i1 @llvm.is.constant.p0(ptr %cast) + ret i1 %ret +} `````````` </details> https://github.com/llvm/llvm-project/pull/102010 _______________________________________________ llvm-branch-commits mailing list llvm-branch-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-branch-commits
