[ASAN] Emit AMDGPU intrinsics only for target triple AMDGPU

[pvellien]

Summary

The functions @llvm.amdgcn.is.private and @llvm.amdgcn.is.shared is emitted for all module which are instrumented regardless of target triple. This patch attempts to fix that behavior by emitting them only for AMDGPU target.

[pvellien]

Retry build.

llvm/lib/Transforms/Instrumentation/AddressSanitizer.cpp

   AsanPtrCmpFunction =
       M.getOrInsertFunction(kAsanPtrCmp, IRB.getVoidTy(), IntptrTy, IntptrTy);
   AsanPtrSubFunction =
       M.getOrInsertFunction(kAsanPtrSub, IRB.getVoidTy(), IntptrTy, IntptrTy);
   if (Mapping.InGlobal)
     AsanShadowGlobal = M.getOrInsertGlobal("__asan_shadow",
                                            ArrayType::get(IRB.getInt8Ty(), 0));
 
-  AMDGPUAddressShared = M.getOrInsertFunction(
-      kAMDGPUAddressSharedName, IRB.getInt1Ty(), IRB.getInt8PtrTy());
-  AMDGPUAddressPrivate = M.getOrInsertFunction(
-      kAMDGPUAddressPrivateName, IRB.getInt1Ty(), IRB.getInt8PtrTy());
+ if (TargetTriple.isAMDGPU()) {
+    AMDGPUAddressShared = M.getOrInsertFunction(
+        kAMDGPUAddressSharedName, IRB.getInt1Ty(), IRB.getInt8PtrTy());
+    AMDGPUAddressPrivate = M.getOrInsertFunction(
+        kAMDGPUAddressPrivateName, IRB.getInt1Ty(), IRB.getInt8PtrTy());
+ }
 }
 
 bool AddressSanitizer::maybeInsertAsanInitAtFunctionEntry(Function &F) {
   // For each NSObject descendant having a +load method, this method is invoked
   // by the ObjC runtime before any of the static constructors is called.
   // Therefore we need to instrument such methods with a call to __asan_init
   // at the beginning in order to initialize our runtime before any access to
   // the shadow memory.

llvm/test/Instrumentation/AddressSanitizer/basic.ll

 ; CHECK: ret void
 define void @test_swifterror_3() sanitize_address {
   %swifterror_addr = alloca swifterror i8*
   store i8* null, i8** %swifterror_addr
   call void @test_swifterror_2(i8** swifterror %swifterror_addr)
   ret void
 }
 
+; CHECK: declare void @__asan_handle_no_return()
+; CHECK-NOT: declare i1 @llvm.amdgcn.is.private
+; CHECK-NOT: declare i1 @llvm.amdgcn.is.shared
+; CHECK: declare void @__asan_init()
+
 ;; ctor/dtor have the nounwind attribute. See uwtable.ll, they additionally have
 ;; the uwtable attribute with the module flag "uwtable".
 ; CHECK: define internal void @asan.module_ctor() #[[#ATTR:]] {{(comdat )?}}{
 ; CHECK: call void @__asan_init()
 
 ; CHECK: attributes #[[#ATTR]] = { nounwind }
 
 ; PROF
 ; CHECK: ![[PROF]] = !{!"branch_weights", i32 1, i32 100000}