Update .global and .align

Author: Vandana2896

llvm/lib/Target/NVPTX/NVPTXAsmPrinter.cpp

@@ -1609,8 +1609,16 @@ void NVPTXAsmPrinter::emitFunctionParamList(const Function *F, raw_ostream &O) {
           int addrSpace = PTy->getAddressSpace();
           if (static_cast<NVPTXTargetMachine &>(TM).getDrvInterface() ==
               NVPTX::CUDA) {
+            // Special handling for pointer arguments to kernel
+            // CUDA kernels assume that pointers are in global address space
+            // See:
+            // https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#parameter-state-space
             assert(addrSpace == 0 && "Invalid address space");
             O << ".ptr .global ";
+            if (I->getParamAlign().valueOrOne() != 1) {
+              Align ParamAlign = I->getParamAlign().value();
+              O << ".align " << ParamAlign.value() << " ";
+            }
           } else {
             switch (addrSpace) {
             default:
@@ -1626,9 +1634,9 @@ void NVPTXAsmPrinter::emitFunctionParamList(const Function *F, raw_ostream &O) {
               O << ".ptr .global ";
               break;
             }
+            Align ParamAlign = I->getParamAlign().valueOrOne();
+            O << ".align " << ParamAlign.value() << " ";
           }
-          Align ParamAlign = I->getParamAlign().valueOrOne();
-          O << ".align " << ParamAlign.value() << " ";
           O << TLI->getParamName(F, paramIndex);
           continue;
         }

llvm/test/CodeGen/NVPTX/kernel-param-align.ll

@@ -4,7 +4,7 @@
 %struct.Large = type { [16 x double] }
 
 ; CHECK: .param .u64 .ptr .global .align 16 func_align_param_0,
-; CHECK: .param .u64 func_align_param_1,
+; CHECK: .param .u64 .ptr .global func_align_param_1,
 ; CHECK: .param .u32 func_align_param_2
 define void @func_align(ptr nocapture readonly align 16 %input, ptr nocapture %out, i32 %n) {
 entry:
@@ -16,8 +16,8 @@ entry:
   ret void
 }
 
-; CHECK: .param .u64 func_param_0,
-; CHECK: .param .u64 func_param_1,
+; CHECK: .param .ptr .global .u64 func_param_0,
+; CHECK: .param .ptr .global .u64 func_param_1,
 ; CHECK: .param .u32 func_param_2
 define void @func(ptr nocapture readonly %input, ptr nocapture %out, i32 %n) {
 entry: