Skip to content

Commit 43b8ae3

Browse files
authored
[AMDGPU][LDS] Pre-Commit tests for 'Fix dynamic LDS interaction with "amdgpu-no-lds-kernel-id" (#107091)
1 parent 55a2473 commit 43b8ae3

File tree

1 file changed

+17
-5
lines changed

1 file changed

+17
-5
lines changed

llvm/test/CodeGen/AMDGPU/lower-module-lds-zero-size-arr.ll

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,18 @@
1-
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --version 4
1+
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --check-globals all --version 5
22
; RUN: opt -S -mtriple=amdgcn-- -passes=amdgpu-lower-module-lds < %s | FileCheck %s
33

44
; This is an extension and should be rejected by the front-end in most cases.
55
; If it goes through, lower it as dynlds.
66

77
@Var0 = linkonce_odr hidden local_unnamed_addr addrspace(3) global [0 x float] poison
88

9-
define void @fn(float %val, i32 %idx) {
9+
;.
10+
; CHECK: @llvm.amdgcn.kernelA.dynlds = external addrspace(3) global [0 x i8], align 4, !absolute_symbol [[META0:![0-9]+]]
11+
; CHECK: @llvm.amdgcn.dynlds.offset.table = internal addrspace(4) constant [1 x i32] [i32 ptrtoint (ptr addrspace(3) @llvm.amdgcn.kernelA.dynlds to i32)]
12+
;.
13+
define void @fn(float %val, i32 %idx) #0 {
1014
; CHECK-LABEL: define void @fn(
11-
; CHECK-SAME: float [[VAL:%.*]], i32 [[IDX:%.*]]) {
15+
; CHECK-SAME: float [[VAL:%.*]], i32 [[IDX:%.*]]) #[[ATTR0:[0-9]+]] {
1216
; CHECK-NEXT: [[TMP1:%.*]] = call i32 @llvm.amdgcn.lds.kernel.id()
1317
; CHECK-NEXT: [[VAR0:%.*]] = getelementptr inbounds [1 x i32], ptr addrspace(4) @llvm.amdgcn.dynlds.offset.table, i32 0, i32 [[TMP1]]
1418
; CHECK-NEXT: [[TMP2:%.*]] = load i32, ptr addrspace(4) [[VAR0]], align 4
@@ -22,16 +26,24 @@ define void @fn(float %val, i32 %idx) {
2226
ret void
2327
}
2428

25-
define amdgpu_kernel void @kernelA(float %val, i32 %idx) {
29+
define amdgpu_kernel void @kernelA(float %val, i32 %idx) #0 {
2630
; CHECK-LABEL: define amdgpu_kernel void @kernelA(
27-
; CHECK-SAME: float [[VAL:%.*]], i32 [[IDX:%.*]]) !llvm.amdgcn.lds.kernel.id [[META1:![0-9]+]] {
31+
; CHECK-SAME: float [[VAL:%.*]], i32 [[IDX:%.*]]) #[[ATTR0]] !llvm.amdgcn.lds.kernel.id [[META1:![0-9]+]] {
2832
; CHECK-NEXT: call void @llvm.donothing() [ "ExplicitUse"(ptr addrspace(3) @llvm.amdgcn.kernelA.dynlds) ]
2933
; CHECK-NEXT: tail call void @fn(float [[VAL]], i32 [[IDX]])
3034
; CHECK-NEXT: ret void
3135
;
3236
tail call void @fn(float %val, i32 %idx)
3337
ret void
3438
}
39+
40+
attributes #0 = { "amdgpu-no-lds-kernel-id" }
41+
42+
;.
43+
; CHECK: attributes #[[ATTR0]] = { "amdgpu-no-lds-kernel-id" }
44+
; CHECK: attributes #[[ATTR1:[0-9]+]] = { nocallback nofree nosync nounwind willreturn memory(none) }
45+
; CHECK: attributes #[[ATTR2:[0-9]+]] = { nocallback nofree nosync nounwind speculatable willreturn memory(none) }
3546
;.
47+
; CHECK: [[META0]] = !{i32 0, i32 1}
3648
; CHECK: [[META1]] = !{i32 0}
3749
;.

0 commit comments

Comments
 (0)