Fixing alignment of input in visa code

Fixing alignment of inputs in visa code for pointers
arguments.

Author: ViacheslavRb

IGC/Compiler/CISACodeGen/OpenCLKernelCodeGen.cpp

@@ -2475,6 +2475,9 @@ namespace IGC
                     // Align on the desired alignment for this argument
                     auto alignment = arg.getAlignment();
 
+                    if (arg.isArgPtrType())
+                        alignment = m_Context->getModule()->getDataLayout().getPointerTypeSize(arg.getArg()->getType());
+
                     // FIXME: move alignment checks to implicit arg creation
                     if ((arg.getArgType() == KernelArg::ArgType::IMPLICIT_LOCAL_IDS ||
                          arg.getArgType() == KernelArg::ArgType::RT_STACK_ID) &&

IGC/Compiler/Optimizer/OpenCLPasses/KernelArgs/KernelArgs.cpp

@@ -686,6 +686,13 @@ bool KernelArg::isBindlessSampler(const Argument* arg, const StringRef typeStr)
     return (typeStr.equals("bindless_sampler_t"));
 }
 
+bool KernelArg::isArgPtrType()
+{
+    return m_argType == ArgType::PTR_LOCAL || m_argType == ArgType::PTR_GLOBAL
+        || m_argType == ArgType::PTR_CONSTANT || m_argType == ArgType::PTR_DEVICE_QUEUE;
+}
+
+
 iOpenCL::DATA_PARAMETER_TOKEN KernelArg::getDataParamToken() const
 {
     auto iter = argTypeTokenMap.find(m_argType);

IGC/Compiler/Optimizer/OpenCLPasses/KernelArgs/KernelArgs.hpp

@@ -221,7 +221,7 @@ namespace IGC
         size_t                          getAlignment()          const;
         bool                            isConstantBuf()         const;
         bool                            needsAllocation()       const;
-        const llvm::Argument* getArg()                const;
+        const llvm::Argument*           getArg()                const;
         unsigned int                    getAssociatedArgNo()    const;
         unsigned int                    getStructArgOffset()    const;
         unsigned int                    getLocationIndex()      const;
@@ -263,6 +263,10 @@ namespace IGC
         static bool isImage(const llvm::Argument* arg, const llvm::StringRef typeStr, ArgType& imageArgType);
         static bool isSampler(const llvm::Argument* arg, const llvm::StringRef typeStr);
 
+        /// @brief  Checks whether the given argument is a sampler
+        /// @return true is argument type is PTR_LOCAL | PTR_GLOBAL | PTR_CONSTANT | PTR_DEVICE_QUEUE
+        bool isArgPtrType();
+
     private:
         /// @brief  Calculates the allocation size needed for the given explicit argument
         /// @param  arg         The kernel argument
@@ -329,7 +333,7 @@ namespace IGC
         /// @brief  Indicates whether the argument is used in calculating the constant buffer length
         bool                            m_isConstantBuf;
         /// @brief  The LLVM argument that represents this kernel argument
-        const llvm::Argument* m_arg;
+        const llvm::Argument*           m_arg;
         /// @brief  The argument number of the associated argument
         ///         For image dimension/BUFFER_OFFSET arguments this will return the argument number
         ///         of the assocaited image.