[llvm] 4f63f97 - AMDGPU: Add sanity test if amdgcn.device.{init|fini} already exists

[Matt Arsenault via llvm-commits]

Author: Matt Arsenault
Date: 2022-12-12T22:18:37-05:00
New Revision: 4f63f9739c086352c9b632671752e78c79c8625b

URL: https://github.com/llvm/llvm-project/commit/4f63f9739c086352c9b632671752e78c79c8625b
DIFF: https://github.com/llvm/llvm-project/commit/4f63f9739c086352c9b632671752e78c79c8625b.diff

LOG: AMDGPU: Add sanity test if amdgcn.device.{init|fini} already exists

Added: 
    llvm/test/CodeGen/AMDGPU/lower-ctor-dtor-existing.ll

Modified: 
    

Removed: 
    


################################################################################
diff  --git a/llvm/test/CodeGen/AMDGPU/lower-ctor-dtor-existing.ll b/llvm/test/CodeGen/AMDGPU/lower-ctor-dtor-existing.ll
new file mode 100644
index 000000000000..5fa23e4ae823
--- /dev/null
+++ b/llvm/test/CodeGen/AMDGPU/lower-ctor-dtor-existing.ll
@@ -0,0 +1,64 @@
+; RUN: opt -S -mtriple=amdgcn-- -passes=amdgpu-lower-ctor-dtor < %s | FileCheck %s
+; RUN: llc -mtriple=amdgcn-amd-amdhsa -mcpu=gfx700 -filetype=obj -o - < %s | llvm-readelf -s - 2>&1 | FileCheck %s -check-prefix=CHECK-VIS
+
+; Make sure there's no crash or error if amdgcn.device.init or
+; amdgcn.device.fini already exist.
+
+ at llvm.global_ctors = appending addrspace(1) global [1 x { i32, ptr, ptr }] [{ i32, ptr, ptr } { i32 1, ptr @foo, ptr null }]
+ at llvm.global_dtors = appending addrspace(1) global [1 x { i32, ptr, ptr }] [{ i32, ptr, ptr } { i32 1, ptr @bar, ptr null }]
+
+; CHECK-NOT: @llvm.global_ctors
+; CHECK-NOT: @llvm.global_dtors
+
+; CHECK-LABEL: amdgpu_kernel void @amdgcn.device.init() #0
+; CHECK-NEXT: store
+; CHECK-NEXT: ret void
+
+; CHECK-LABEL: amdgpu_kernel void @amdgcn.device.fini() #1
+; CHECK-NEXT: store
+; CHECK-NEXT: ret void
+
+
+; CHECK-LABEL: amdgpu_kernel void @amdgcn.device.init.1() #0
+; CHECK-NEXT: call void @foo
+; CHECK-NEXT: ret void
+
+; CHECK-LABEL: amdgpu_kernel void @amdgcn.device.fini.2() #1
+; CHECK-NEXT: call void @bar
+; CHECK-NEXT: ret void
+
+; CHECK-NOT: amdgcn.device.
+
+; CHECK-VIS: FUNC   GLOBAL PROTECTED {{.*}} amdgcn.device.init{{$}}
+; CHECK-VIS: OBJECT GLOBAL DEFAULT {{.*}} amdgcn.device.init.kd{{$}}
+; CHECK-VIS: FUNC   GLOBAL PROTECTED {{.*}} amdgcn.device.fini{{$}}
+; CHECK-VIS: OBJECT   GLOBAL DEFAULT {{.*}} amdgcn.device.fini.kd{{$}}
+
+; CHECK-VIS: FUNC   GLOBAL PROTECTED {{.*}} amdgcn.device.init.1{{$}}
+; CHECK-VIS: OBJECT GLOBAL DEFAULT {{.*}} amdgcn.device.init.1.kd{{$}}
+; CHECK-VIS: FUNC   GLOBAL PROTECTED {{.*}} amdgcn.device.fini.2{{$}}
+; CHECK-VIS: OBJECT   GLOBAL DEFAULT {{.*}} amdgcn.device.fini.2.kd{{$}}
+
+
+define internal void @foo() {
+  ret void
+}
+
+define internal void @bar() {
+  ret void
+}
+
+define amdgpu_kernel void @amdgcn.device.init() #0 {
+  store volatile i32 1, ptr addrspace(1) null
+  ret void
+}
+
+define amdgpu_kernel void @amdgcn.device.fini() #1 {
+  store volatile i32 0, ptr addrspace(1) null
+  ret void
+}
+
+attributes #0 = { "device-init" }
+attributes #1 = { "device-fini" }
+; CHECK: attributes #0 = { "device-init" }
+; CHECK: attributes #1 = { "device-fini" }