This is a partial cleanup to centralize the initialization and update decisions for AAs. Lifting the burdon and boilerplate on users and making it harder to accidentally perform unsound deductions. The two static helpers show how we can lift the decisions to generate an AA into the Attributor, avoiding trivial AAs that just cost us compile time and maintenance code (to check for pre-conditions).
60 lines
2.6 KiB
LLVM
60 lines
2.6 KiB
LLVM
; NOTE: Assertions have been autogenerated by utils/update_test_checks.py UTC_ARGS: --function-signature --check-globals
|
|
; RUN: opt -S -mtriple=amdgcn-amd- -amdgpu-attributor %s | FileCheck %s
|
|
|
|
@x = global i32 0
|
|
|
|
; Propagate the uniform-work-group-attribute from the kernel to callee if it doesn't have it
|
|
;.
|
|
; CHECK: @[[X:[a-zA-Z0-9_$"\\.-]+]] = global i32 0
|
|
;.
|
|
define void @func() #0 {
|
|
; CHECK-LABEL: define {{[^@]+}}@func
|
|
; CHECK-SAME: () #[[ATTR0:[0-9]+]] {
|
|
; CHECK-NEXT: store i32 0, ptr @x, align 4
|
|
; CHECK-NEXT: ret void
|
|
;
|
|
store i32 0, ptr @x
|
|
ret void
|
|
}
|
|
|
|
define amdgpu_kernel void @kernel1() #1 {
|
|
; CHECK-LABEL: define {{[^@]+}}@kernel1
|
|
; CHECK-SAME: () #[[ATTR1:[0-9]+]] {
|
|
; CHECK-NEXT: call void @func()
|
|
; CHECK-NEXT: ret void
|
|
;
|
|
call void @func()
|
|
ret void
|
|
}
|
|
|
|
; External declaration of a function
|
|
define weak_odr void @weak_func() #0 {
|
|
; CHECK-LABEL: define {{[^@]+}}@weak_func
|
|
; CHECK-SAME: () #[[ATTR2:[0-9]+]] {
|
|
; CHECK-NEXT: store i32 0, ptr @x, align 4
|
|
; CHECK-NEXT: ret void
|
|
;
|
|
store i32 0, ptr @x
|
|
ret void
|
|
}
|
|
|
|
define amdgpu_kernel void @kernel2() #2 {
|
|
; CHECK-LABEL: define {{[^@]+}}@kernel2
|
|
; CHECK-SAME: () #[[ATTR3:[0-9]+]] {
|
|
; CHECK-NEXT: call void @weak_func()
|
|
; CHECK-NEXT: ret void
|
|
;
|
|
call void @weak_func()
|
|
ret void
|
|
}
|
|
|
|
attributes #0 = { nounwind }
|
|
attributes #1 = { "uniform-work-group-size"="false" }
|
|
attributes #2 = { "uniform-work-group-size"="true" }
|
|
;.
|
|
; CHECK: attributes #[[ATTR0]] = { nounwind "amdgpu-no-completion-action" "amdgpu-no-default-queue" "amdgpu-no-dispatch-id" "amdgpu-no-dispatch-ptr" "amdgpu-no-heap-ptr" "amdgpu-no-hostcall-ptr" "amdgpu-no-implicitarg-ptr" "amdgpu-no-lds-kernel-id" "amdgpu-no-multigrid-sync-arg" "amdgpu-no-queue-ptr" "amdgpu-no-workgroup-id-x" "amdgpu-no-workgroup-id-y" "amdgpu-no-workgroup-id-z" "amdgpu-no-workitem-id-x" "amdgpu-no-workitem-id-y" "amdgpu-no-workitem-id-z" "amdgpu-waves-per-eu"="4,10" "uniform-work-group-size"="false" }
|
|
; CHECK: attributes #[[ATTR1]] = { "amdgpu-no-completion-action" "amdgpu-no-default-queue" "amdgpu-no-dispatch-id" "amdgpu-no-dispatch-ptr" "amdgpu-no-heap-ptr" "amdgpu-no-hostcall-ptr" "amdgpu-no-implicitarg-ptr" "amdgpu-no-lds-kernel-id" "amdgpu-no-multigrid-sync-arg" "amdgpu-no-queue-ptr" "amdgpu-no-workgroup-id-x" "amdgpu-no-workgroup-id-y" "amdgpu-no-workgroup-id-z" "amdgpu-no-workitem-id-x" "amdgpu-no-workitem-id-y" "amdgpu-no-workitem-id-z" "uniform-work-group-size"="false" }
|
|
; CHECK: attributes #[[ATTR2]] = { nounwind "amdgpu-waves-per-eu"="4,10" "uniform-work-group-size"="false" }
|
|
; CHECK: attributes #[[ATTR3]] = { "uniform-work-group-size"="true" }
|
|
;.
|