forked from OSchip/llvm-project
269 lines
8.6 KiB
LLVM
269 lines
8.6 KiB
LLVM
; RUN: opt -S -mtriple=amdgcn-- -amdgpu-codegenprepare -amdgpu-codegenprepare-widen-constant-loads < %s | FileCheck -check-prefix=OPT %s
|
|
|
|
declare i8 addrspace(4)* @llvm.amdgcn.dispatch.ptr() #0
|
|
|
|
; OPT-LABEL: @constant_load_i1
|
|
; OPT: load i1
|
|
; OPT-NEXT: store i1
|
|
define amdgpu_kernel void @constant_load_i1(i1 addrspace(1)* %out, i1 addrspace(4)* %in) #0 {
|
|
%val = load i1, i1 addrspace(4)* %in
|
|
store i1 %val, i1 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i1_align2
|
|
; OPT: load i1
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_i1_align2(i1 addrspace(1)* %out, i1 addrspace(4)* %in) #0 {
|
|
%val = load i1, i1 addrspace(4)* %in, align 2
|
|
store i1 %val, i1 addrspace(1)* %out, align 2
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i1_align4
|
|
; OPT: bitcast
|
|
; OPT-NEXT: load i32
|
|
; OPT-NEXT: trunc
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_i1_align4(i1 addrspace(1)* %out, i1 addrspace(4)* %in) #0 {
|
|
%val = load i1, i1 addrspace(4)* %in, align 4
|
|
store i1 %val, i1 addrspace(1)* %out, align 4
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i8
|
|
; OPT: load i8
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_i8(i8 addrspace(1)* %out, i8 addrspace(4)* %in) #0 {
|
|
%val = load i8, i8 addrspace(4)* %in
|
|
store i8 %val, i8 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i8_align2
|
|
; OPT: load i8
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_i8_align2(i8 addrspace(1)* %out, i8 addrspace(4)* %in) #0 {
|
|
%val = load i8, i8 addrspace(4)* %in, align 2
|
|
store i8 %val, i8 addrspace(1)* %out, align 2
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i8align4
|
|
; OPT: bitcast
|
|
; OPT-NEXT: load i32
|
|
; OPT-NEXT: trunc
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_i8align4(i8 addrspace(1)* %out, i8 addrspace(4)* %in) #0 {
|
|
%val = load i8, i8 addrspace(4)* %in, align 4
|
|
store i8 %val, i8 addrspace(1)* %out, align 4
|
|
ret void
|
|
}
|
|
|
|
|
|
; OPT-LABEL: @constant_load_v2i8
|
|
; OPT: load <2 x i8>
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_v2i8(<2 x i8> addrspace(1)* %out, <2 x i8> addrspace(4)* %in) #0 {
|
|
%ld = load <2 x i8>, <2 x i8> addrspace(4)* %in
|
|
store <2 x i8> %ld, <2 x i8> addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_v2i8_align4
|
|
; OPT: bitcast
|
|
; OPT-NEXT: load i32
|
|
; OPT-NEXT: trunc
|
|
; OPT-NEXT: bitcast
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_v2i8_align4(<2 x i8> addrspace(1)* %out, <2 x i8> addrspace(4)* %in) #0 {
|
|
%ld = load <2 x i8>, <2 x i8> addrspace(4)* %in, align 4
|
|
store <2 x i8> %ld, <2 x i8> addrspace(1)* %out, align 4
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_v3i8
|
|
; OPT: bitcast <3 x i8>
|
|
; OPT-NEXT: load i32, i32 addrspace(4)
|
|
; OPT-NEXT: trunc i32
|
|
; OPT-NEXT: bitcast i24
|
|
; OPT-NEXT: store <3 x i8>
|
|
define amdgpu_kernel void @constant_load_v3i8(<3 x i8> addrspace(1)* %out, <3 x i8> addrspace(4)* %in) #0 {
|
|
%ld = load <3 x i8>, <3 x i8> addrspace(4)* %in
|
|
store <3 x i8> %ld, <3 x i8> addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_v3i8_align4
|
|
; OPT: bitcast <3 x i8>
|
|
; OPT-NEXT: load i32, i32 addrspace(4)
|
|
; OPT-NEXT: trunc i32
|
|
; OPT-NEXT: bitcast i24
|
|
; OPT-NEXT: store <3 x i8>
|
|
define amdgpu_kernel void @constant_load_v3i8_align4(<3 x i8> addrspace(1)* %out, <3 x i8> addrspace(4)* %in) #0 {
|
|
%ld = load <3 x i8>, <3 x i8> addrspace(4)* %in, align 4
|
|
store <3 x i8> %ld, <3 x i8> addrspace(1)* %out, align 4
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i16
|
|
; OPT: load i16
|
|
; OPT: sext
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_i16(i32 addrspace(1)* %out, i16 addrspace(4)* %in) #0 {
|
|
%ld = load i16, i16 addrspace(4)* %in
|
|
%ext = sext i16 %ld to i32
|
|
store i32 %ext, i32 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i16_align4
|
|
; OPT: bitcast
|
|
; OPT-NEXT: load i32
|
|
; OPT-NEXT: trunc
|
|
; OPT-NEXT: sext
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_i16_align4(i32 addrspace(1)* %out, i16 addrspace(4)* %in) #0 {
|
|
%ld = load i16, i16 addrspace(4)* %in, align 4
|
|
%ext = sext i16 %ld to i32
|
|
store i32 %ext, i32 addrspace(1)* %out, align 4
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_f16
|
|
; OPT: load half
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_f16(half addrspace(1)* %out, half addrspace(4)* %in) #0 {
|
|
%ld = load half, half addrspace(4)* %in
|
|
store half %ld, half addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_v2f16
|
|
; OPT: load <2 x half>
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_v2f16(<2 x half> addrspace(1)* %out, <2 x half> addrspace(4)* %in) #0 {
|
|
%ld = load <2 x half>, <2 x half> addrspace(4)* %in
|
|
store <2 x half> %ld, <2 x half> addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @load_volatile
|
|
; OPT: load volatile i16
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @load_volatile(i16 addrspace(1)* %out, i16 addrspace(4)* %in) {
|
|
%a = load volatile i16, i16 addrspace(4)* %in
|
|
store i16 %a, i16 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_v2i8_volatile
|
|
; OPT: load volatile <2 x i8>
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_v2i8_volatile(<2 x i8> addrspace(1)* %out, <2 x i8> addrspace(4)* %in) #0 {
|
|
%ld = load volatile <2 x i8>, <2 x i8> addrspace(4)* %in
|
|
store <2 x i8> %ld, <2 x i8> addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_v2i8_addrspace1
|
|
; OPT: load <2 x i8>
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @constant_load_v2i8_addrspace1(<2 x i8> addrspace(1)* %out, <2 x i8> addrspace(1)* %in) #0 {
|
|
%ld = load <2 x i8>, <2 x i8> addrspace(1)* %in
|
|
store <2 x i8> %ld, <2 x i8> addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @use_dispatch_ptr
|
|
; OPT: bitcast
|
|
; OPT-NEXT: load i32
|
|
; OPT-NEXT: trunc
|
|
; OPT-NEXT: zext
|
|
; OPT-NEXT: store
|
|
define amdgpu_kernel void @use_dispatch_ptr(i32 addrspace(1)* %ptr) #1 {
|
|
%dispatch.ptr = call i8 addrspace(4)* @llvm.amdgcn.dispatch.ptr()
|
|
%val = load i8, i8 addrspace(4)* %dispatch.ptr, align 4
|
|
%ld = zext i8 %val to i32
|
|
store i32 %ld, i32 addrspace(1)* %ptr
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i16_align4_range(
|
|
; OPT: load i32, i32 addrspace(4)* %1, align 4, !range !0
|
|
define amdgpu_kernel void @constant_load_i16_align4_range(i32 addrspace(1)* %out, i16 addrspace(4)* %in) #0 {
|
|
%ld = load i16, i16 addrspace(4)* %in, align 4, !range !0
|
|
%ext = sext i16 %ld to i32
|
|
store i32 %ext, i32 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i16_align4_range_max(
|
|
; OPT: load i32, i32 addrspace(4)* %1, align 4, !range !0
|
|
define amdgpu_kernel void @constant_load_i16_align4_range_max(i32 addrspace(1)* %out, i16 addrspace(4)* %in) #0 {
|
|
%ld = load i16, i16 addrspace(4)* %in, align 4, !range !1
|
|
%ext = sext i16 %ld to i32
|
|
store i32 %ext, i32 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i16_align4_complex_range(
|
|
; OPT: load i32, i32 addrspace(4)* %1, align 4, !range !1
|
|
define amdgpu_kernel void @constant_load_i16_align4_complex_range(i32 addrspace(1)* %out, i16 addrspace(4)* %in) #0 {
|
|
%ld = load i16, i16 addrspace(4)* %in, align 4, !range !2
|
|
%ext = sext i16 %ld to i32
|
|
store i32 %ext, i32 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i16_align4_range_from_0(
|
|
; OPT: load i32, i32 addrspace(4)* %1, align 4{{$}}
|
|
define amdgpu_kernel void @constant_load_i16_align4_range_from_0(i32 addrspace(1)* %out, i16 addrspace(4)* %in) #0 {
|
|
%ld = load i16, i16 addrspace(4)* %in, align 4, !range !3
|
|
%ext = sext i16 %ld to i32
|
|
store i32 %ext, i32 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i16_align4_range_from_neg(
|
|
; OPT: load i32, i32 addrspace(4)* %1, align 4, !range !2
|
|
define amdgpu_kernel void @constant_load_i16_align4_range_from_neg(i32 addrspace(1)* %out, i16 addrspace(4)* %in) #0 {
|
|
%ld = load i16, i16 addrspace(4)* %in, align 4, !range !4
|
|
%ext = sext i16 %ld to i32
|
|
store i32 %ext, i32 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i16_align4_range_from_neg_to_0(
|
|
; OPT: load i32, i32 addrspace(4)* %1, align 4, !range !2
|
|
define amdgpu_kernel void @constant_load_i16_align4_range_from_neg_to_0(i32 addrspace(1)* %out, i16 addrspace(4)* %in) #0 {
|
|
%ld = load i16, i16 addrspace(4)* %in, align 4, !range !5
|
|
%ext = sext i16 %ld to i32
|
|
store i32 %ext, i32 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
; OPT-LABEL: @constant_load_i16_align4_invariant
|
|
; OPT: load i32, i32 addrspace(4)* %1, align 4, !invariant.load !3
|
|
define amdgpu_kernel void @constant_load_i16_align4_invariant(i32 addrspace(1)* %out, i16 addrspace(4)* %in) #0 {
|
|
%ld = load i16, i16 addrspace(4)* %in, align 4, !invariant.load !6
|
|
%ext = sext i16 %ld to i32
|
|
store i32 %ext, i32 addrspace(1)* %out
|
|
ret void
|
|
}
|
|
|
|
attributes #0 = { nounwind }
|
|
|
|
; OPT: !0 = !{i32 5, i32 0}
|
|
; OPT: !1 = !{i32 8, i32 0}
|
|
; OPT: !2 = !{i32 65520, i32 0}
|
|
; OPT: !3 = !{}
|
|
|
|
!0 = !{i16 5, i16 500}
|
|
!1 = !{i16 5, i16 -1}
|
|
!2 = !{i16 8, i16 12, i16 42, i16 99}
|
|
!3 = !{i16 0, i16 255}
|
|
!4 = !{i16 -16, i16 16}
|
|
!5 = !{i16 -16, i16 0}
|
|
!6 = !{}
|