PTX: Add .align tests to stack object test file

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@140537 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/test/CodeGen/PTX/stack-object.ll b/test/CodeGen/PTX/stack-object.ll
index 07cc0ab..aab7f51 100644
--- a/test/CodeGen/PTX/stack-object.ll
+++ b/test/CodeGen/PTX/stack-object.ll
@@ -1,8 +1,17 @@
 ; RUN: llc < %s -march=ptx32 -mattr=sm20 | FileCheck %s
 
 define ptx_device float @stack1(float %a) {
-  ; CHECK: .local .b32 __local0;
-  %a.2 = alloca float
+  ; CHECK: .local .align 4 .b32 __local0;
+  %a.2 = alloca float, align 4
+  ; CHECK: st.local.f32 [__local0], %f0
+  store float %a, float* %a.2
+  %a.3 = load float* %a.2
+  ret float %a.3
+}
+
+define ptx_device float @stack1_align8(float %a) {
+  ; CHECK: .local .align 8 .b32 __local0;
+  %a.2 = alloca float, align 8
   ; CHECK: st.local.f32 [__local0], %f0
   store float %a, float* %a.2
   %a.3 = load float* %a.2