Fix printf 64bit pointertype

Given example like:
  %ptrBC = bitcast i8 addrspace(1)* %printfBuffer to i32 addrspace(1)*
  %write_offset = call i32 @__builtin_IB_atomic_add_global_i32(i32 addrspace(1)* %ptrBC, i32 12)
Whether ptrBC is 64bit should depends on its type instead of PointerSize of DataLayout.

Author: Jay-Jiewu-Lu

IGC/Compiler/Optimizer/OpenCLPasses/Atomics/ResolveOCLAtomics.cpp

@@ -26,6 +26,7 @@ using namespace IGC;
 #define PASS_CFG_ONLY false
 #define PASS_ANALYSIS false
 IGC_INITIALIZE_PASS_BEGIN(ResolveOCLAtomics, PASS_FLAG, PASS_DESCRIPTION, PASS_CFG_ONLY, PASS_ANALYSIS)
+IGC_INITIALIZE_PASS_DEPENDENCY(CodeGenContextWrapper)
 IGC_INITIALIZE_PASS_END(ResolveOCLAtomics, PASS_FLAG, PASS_DESCRIPTION, PASS_CFG_ONLY, PASS_ANALYSIS)
 
 char ResolveOCLAtomics::ID = 0;
@@ -77,24 +78,13 @@ void ResolveOCLAtomics::initOCLAtomicsMap()
 
 bool ResolveOCLAtomics::runOnModule(Module& M)
 {
+    m_CGCtx = getAnalysis<CodeGenContextWrapper>().getCodeGenContext();
     m_pModule  = (IGCLLVM::Module*)&M;
     m_Int32Ty = Type::getInt32Ty(m_pModule->getContext());
 
     llvm::IGCIRBuilder<> builder(M.getContext());
     m_builder = &builder;
 
-    int pointerSize = M.getDataLayout().getPointerSizeInBits();
-    IGC_ASSERT(pointerSize == 64 || pointerSize == 32);
-
-    if (pointerSize == 64)
-    {
-        m_64bitPointer = true;
-    }
-    else
-    {
-        m_64bitPointer = false;
-    }
-
     m_changed = false;
 
     // Visit all call instructions in the function F.
@@ -147,7 +137,10 @@ void ResolveOCLAtomics::processOCLAtomic(CallInst& callInst, AtomicOp op, Buffer
         callInst.getOperand(1);
 
     const bool floatArgs = !noSources && src0->getType()->isFloatingPointTy();
-    const bool is64bit = m_64bitPointer && bufType != SLM;
+
+    Value* dstBuffer = callInst.getOperand(0);
+    PointerType* PtrTy = dyn_cast<PointerType>(dstBuffer->getType());
+    const bool is64bit = PtrTy && isA64Ptr(PtrTy, m_CGCtx) && bufType != SLM;
 
     // Cmpxchg intrinsic has 2 sources.
     if (op == EATOMIC_CMPXCHG ||
@@ -188,7 +181,6 @@ void ResolveOCLAtomics::processOCLAtomic(CallInst& callInst, AtomicOp op, Buffer
         }
     }
 
-    Value* dstBuffer = callInst.getOperand(0);
     Value* dst = callInst.getOperand(0);
 
     // We will use 64-bit dst only for 64-bit global pointers.

IGC/Compiler/Optimizer/OpenCLPasses/Atomics/ResolveOCLAtomics.hpp

@@ -52,6 +52,11 @@ namespace IGC
         return "ResolveOCLAtomics";
     }
 
+    virtual void getAnalysisUsage(llvm::AnalysisUsage& AU) const override
+    {
+        AU.addRequired<CodeGenContextWrapper>();
+    }
+
     // Entry point of the pass.
         virtual bool runOnModule(llvm::Module& M) override;
 
@@ -61,10 +66,10 @@ namespace IGC
     static const unsigned int ATTR_BUFFER_TYPE_SHIFT = 8;
 
     protected:
-        IGCLLVM::Module* m_pModule;
-        llvm::IGCIRBuilder<>* m_builder;
-        llvm::IntegerType* m_Int32Ty;
-        bool               m_64bitPointer;
+        CodeGenContext* m_CGCtx = nullptr;
+        IGCLLVM::Module* m_pModule = nullptr;
+        llvm::IGCIRBuilder<>* m_builder = nullptr;
+        llvm::IntegerType* m_Int32Ty = nullptr;
 
         // A map that keeps attributes for each "__builtin_IB_atomic_*" function name.
         std::map<llvm::StringRef, OCLAtomicAttrs>  m_AtomicDescMap;