Files
TheRock/compiler/spirv-llvm-translator/test/transcoding/OpImageQuerySize.ll
T
derek.holloway 7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

281 lines
13 KiB
LLVM

; ModuleID = '<stdin>'
target datalayout = "e-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-v256:256-v512:512-v1024:1024"
target triple = "spir64-unknown-unknown"
; RUN: llvm-as %s -o %t.bc
; RUN: llvm-spirv %t.bc -spirv-text -o %t.txt
; RUN: FileCheck < %t.txt %s --check-prefix=CHECK-SPIRV
; RUN: llvm-spirv %t.bc -o %t.spv
; RUN: spirv-val %t.spv
; RUN: llvm-spirv -r %t.spv -o %t.bc
; RUN: llvm-dis < %t.bc | FileCheck %s
; RUN: %if spirv-backend %{ llc -O0 -mtriple=spirv64-unknown-unknown -filetype=obj %s -o %t.llc.spv %}
; RUN: %if spirv-backend %{ llvm-spirv -r %t.llc.spv -o %t.llc.rev.bc %}
; RUN: %if spirv-backend %{ llvm-dis %t.llc.rev.bc -o %t.llc.rev.ll %}
; RUN: %if spirv-backend %{ FileCheck %s < %t.llc.rev.ll %}
; Check conversion of get_image_width, get_image_height, get_image_depth,
; get_image_array_size, and get_image_dim OCL built-ins.
; In general the SPIR-V reader converts OpImageQuerySize into get_image_dim
; and subsequent extract or shufflevector instructions. Unfortunately there is
; no get_image_dim for 1D images and get_image_dim cannot replace get_image_array_size
; CHECK-SPIRV: 10 TypeImage [[ArrayTypeID:[0-9]+]] {{[0-9]+}} 0 0 1 0 0 0 0
%opencl.image1d_ro_t = type opaque
%opencl.image1d_buffer_ro_t = type opaque
%opencl.image1d_array_ro_t = type opaque
%opencl.image2d_ro_t = type opaque
%opencl.image2d_depth_ro_t = type opaque
%opencl.image2d_array_ro_t = type opaque
%opencl.image2d_array_depth_ro_t = type opaque
%opencl.image3d_ro_t = type opaque
; CHECK-LABEL: define {{.*}} @test_image1d
; CHECK: call {{.*}} @_Z15get_image_width14ocl_image1d_ro
; CHECK: call {{.*}} @_Z15get_image_width21ocl_image1d_buffer_ro
; CHECK: call {{.*}} @_Z15get_image_width20ocl_image1d_array_ro
; CHECK: insertelement <2 x i32> {{.*}} 0
; CHECK: call {{.*}} i64 @_Z20get_image_array_size20ocl_image1d_array_ro
; CHECK: trunc i64 {{.*}} to i32
; CHECK: insertelement <2 x i32> {{.*}} 1
; CHECK: call {{.*}} i64 @_Z20get_image_array_size20ocl_image1d_array_ro
; CHECK-SPIRV: 3 FunctionParameter [[ArrayTypeID]] [[ArrayVarID:[0-9]+]]
; CHECK-SPIRV: ImageQuerySizeLod {{[0-9]+}} {{[0-9]+}} [[ArrayVarID]]
; CHECK-SPIRV-NOT: {{[0-9]*}} ExtInst {{[0-9]*}} {{[0-9]*}} {{[0-9]*}} get_image_array_size
; Function Attrs: nounwind
define spir_kernel void @test_image1d(ptr addrspace(1) captures(none) %sizes, ptr addrspace(1) %img, ptr addrspace(1) %buffer, ptr addrspace(1) %array) #0 !kernel_arg_addr_space !1 !kernel_arg_access_qual !2 !kernel_arg_type !3 !kernel_arg_base_type !5 !kernel_arg_type_qual !4 {
%1 = tail call spir_func i32 @_Z15get_image_width14ocl_image1d_ro(ptr addrspace(1) %img) #1
%2 = tail call spir_func i32 @_Z15get_image_width21ocl_image1d_buffer_ro(ptr addrspace(1) %buffer) #1
%3 = tail call spir_func i32 @_Z15get_image_width20ocl_image1d_array_ro(ptr addrspace(1) %array) #1
%4 = tail call spir_func i64 @_Z20get_image_array_size20ocl_image1d_array_ro(ptr addrspace(1) %array) #1
%5 = trunc i64 %4 to i32
%6 = add nsw i32 %2, %1
%7 = add nsw i32 %6, %3
%8 = add nsw i32 %7, %5
store i32 %8, ptr addrspace(1) %sizes, align 4, !tbaa !22
ret void
}
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z15get_image_width14ocl_image1d_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z15get_image_width21ocl_image1d_buffer_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z15get_image_width20ocl_image1d_array_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i64 @_Z20get_image_array_size20ocl_image1d_array_ro(ptr addrspace(1)) #1
; CHECK-LABEL: define {{.*}} @test_image2d
; CHECK: call {{.*}} @_Z13get_image_dim14ocl_image2d_ro
; CHECK: extractelement <2 x i32> {{.*}} 0
; CHECK: call {{.*}} @_Z13get_image_dim14ocl_image2d_ro
; CHECK: extractelement <2 x i32> {{.*}} 1
; CHECK: call {{.*}} @_Z13get_image_dim14ocl_image2d_ro
; CHECK: call {{.*}} @_Z13get_image_dim20ocl_image2d_array_ro
; CHECK: shufflevector <2 x i32> {{.*}} <3 x i32>
; CHECK: call {{.*}} i64 @_Z20get_image_array_size20ocl_image2d_array_ro
; CHECK: trunc i64 {{.*}} to i32
; CHECK: insertelement <3 x i32> {{.*}} i32 2
; CHECK: extractelement <3 x i32> {{.*}} 0
; CHECK: call {{.*}} @_Z13get_image_dim20ocl_image2d_array_ro
; CHECK: shufflevector <2 x i32> {{.*}} <3 x i32>
; CHECK: call {{.*}} i64 @_Z20get_image_array_size20ocl_image2d_array_ro
; CHECK: trunc i64 {{.*}} to i32
; CHECK: insertelement <3 x i32> {{.*}} i32 2
; CHECK: extractelement <3 x i32> {{.*}} 1
; CHECK: call {{.*}} @_Z20get_image_array_size20ocl_image2d_array_ro
; CHECK: call {{.*}} @_Z13get_image_dim20ocl_image2d_array_ro
; CHECK: shufflevector <2 x i32> {{.*}} <3 x i32>
; CHECK: call {{.*}} i64 @_Z20get_image_array_size20ocl_image2d_array_ro
; CHECK: trunc i64 {{.*}} to i32
; CHECK: insertelement <3 x i32> {{.*}} i32 2
; CHECK: shufflevector <3 x i32> {{.*}} <2 x i32>
; Function Attrs: nounwind
define spir_kernel void @test_image2d(ptr addrspace(1) captures(none) %sizes, ptr addrspace(1) %img, ptr addrspace(1) captures(none) %img_depth, ptr addrspace(1) %array, ptr addrspace(1) captures(none) %array_depth) #0 !kernel_arg_addr_space !7 !kernel_arg_access_qual !8 !kernel_arg_type !9 !kernel_arg_base_type !11 !kernel_arg_type_qual !10 {
%1 = tail call spir_func i32 @_Z15get_image_width14ocl_image2d_ro(ptr addrspace(1) %img) #1
%2 = tail call spir_func i32 @_Z16get_image_height14ocl_image2d_ro(ptr addrspace(1) %img) #1
%3 = tail call spir_func <2 x i32> @_Z13get_image_dim14ocl_image2d_ro(ptr addrspace(1) %img) #1
%4 = tail call spir_func i32 @_Z15get_image_width20ocl_image2d_array_ro(ptr addrspace(1) %array) #1
%5 = tail call spir_func i32 @_Z16get_image_height20ocl_image2d_array_ro(ptr addrspace(1) %array) #1
%6 = tail call spir_func i64 @_Z20get_image_array_size20ocl_image2d_array_ro(ptr addrspace(1) %array) #1
%7 = trunc i64 %6 to i32
%8 = tail call spir_func <2 x i32> @_Z13get_image_dim20ocl_image2d_array_ro(ptr addrspace(1) %array) #1
%9 = add nsw i32 %2, %1
%10 = extractelement <2 x i32> %3, i32 0
%11 = add nsw i32 %9, %10
%12 = extractelement <2 x i32> %3, i32 1
%13 = add nsw i32 %11, %12
%14 = add nsw i32 %13, %4
%15 = add nsw i32 %14, %5
%16 = add nsw i32 %15, %7
%17 = extractelement <2 x i32> %8, i32 0
%18 = add nsw i32 %16, %17
%19 = extractelement <2 x i32> %8, i32 1
%20 = add nsw i32 %18, %19
store i32 %20, ptr addrspace(1) %sizes, align 4, !tbaa !22
ret void
}
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z15get_image_width14ocl_image2d_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z16get_image_height14ocl_image2d_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func <2 x i32> @_Z13get_image_dim14ocl_image2d_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z15get_image_width20ocl_image2d_array_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z16get_image_height20ocl_image2d_array_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i64 @_Z20get_image_array_size20ocl_image2d_array_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func <2 x i32> @_Z13get_image_dim20ocl_image2d_array_ro(ptr addrspace(1)) #1
; CHECK-LABEL: define {{.*}} @test_image3d
; CHECK: call {{.*}} @_Z13get_image_dim14ocl_image3d_ro
; CHECK: shufflevector <4 x i32> {{.*}} <3 x i32>
; CHECK: extractelement <3 x i32> {{.*}} 0
; CHECK: call {{.*}} @_Z13get_image_dim14ocl_image3d_ro
; CHECK: shufflevector <4 x i32> {{.*}} <3 x i32>
; CHECK: extractelement <3 x i32> {{.*}} 1
; CHECK: call {{.*}} @_Z13get_image_dim14ocl_image3d_ro
; CHECK: shufflevector <4 x i32> {{.*}} <3 x i32>
; CHECK: extractelement <3 x i32> {{.*}} 2
; CHECK: call {{.*}} @_Z13get_image_dim14ocl_image3d_ro
; CHECK: shufflevector <4 x i32> {{.*}} <3 x i32>
; CHECK: shufflevector <3 x i32> {{.*}} <4 x i32>
; Function Attrs: nounwind
define spir_kernel void @test_image3d(ptr addrspace(1) captures(none) %sizes, ptr addrspace(1) %img) #0 !kernel_arg_addr_space !13 !kernel_arg_access_qual !14 !kernel_arg_type !15 !kernel_arg_base_type !17 !kernel_arg_type_qual !16 {
%1 = tail call spir_func i32 @_Z15get_image_width14ocl_image3d_ro(ptr addrspace(1) %img) #1
%2 = tail call spir_func i32 @_Z16get_image_height14ocl_image3d_ro(ptr addrspace(1) %img) #1
%3 = tail call spir_func i32 @_Z15get_image_depth14ocl_image3d_ro(ptr addrspace(1) %img) #1
%4 = tail call spir_func <4 x i32> @_Z13get_image_dim14ocl_image3d_ro(ptr addrspace(1) %img) #1
%5 = add nsw i32 %2, %1
%6 = add nsw i32 %5, %3
%7 = extractelement <4 x i32> %4, i32 0
%8 = add nsw i32 %6, %7
%9 = extractelement <4 x i32> %4, i32 1
%10 = add nsw i32 %8, %9
%11 = extractelement <4 x i32> %4, i32 2
%12 = add nsw i32 %10, %11
%13 = extractelement <4 x i32> %4, i32 3
%14 = add nsw i32 %12, %13
store i32 %14, ptr addrspace(1) %sizes, align 4, !tbaa !22
ret void
}
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z15get_image_width14ocl_image3d_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z16get_image_height14ocl_image3d_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z15get_image_depth14ocl_image3d_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func <4 x i32> @_Z13get_image_dim14ocl_image3d_ro(ptr addrspace(1)) #1
; CHECK-LABEL: define {{.*}} @test_image2d_array_depth_t
; CHECK: call {{.*}} <2 x i32> @_Z13get_image_dim26ocl_image2d_array_depth_ro
; CHECK: shufflevector <2 x i32>
; CHECK: call {{.*}} i64 @_Z20get_image_array_size26ocl_image2d_array_depth_ro
; CHECK: trunc i64 {{.*}} to i32
; CHECK: insertelement <3 x i32> {{.*}} 2
; CHECK: extractelement <3 x i32> {{.*}} 0
; CHECK: call {{.*}} <2 x i32> @_Z13get_image_dim26ocl_image2d_array_depth_ro
; CHECK: shufflevector <2 x i32>
; CHECK: call {{.*}} i64 @_Z20get_image_array_size26ocl_image2d_array_depth_ro
; CHECK: trunc i64 {{.*}} to i32
; CHECK: insertelement <3 x i32> {{.*}} 2
; CHECK: extractelement <3 x i32> {{.*}} 1
; Function Attrs: nounwind
define spir_kernel void @test_image2d_array_depth_t(ptr addrspace(1) captures(none) %sizes, ptr addrspace(1) %array) #0 !kernel_arg_addr_space !27 !kernel_arg_access_qual !28 !kernel_arg_type !29 !kernel_arg_base_type !31 !kernel_arg_type_qual !30 {
%1 = tail call spir_func i32 @_Z15get_image_width26ocl_image2d_array_depth_ro(ptr addrspace(1) %array) #1
%2 = tail call spir_func i32 @_Z16get_image_height26ocl_image2d_array_depth_ro(ptr addrspace(1) %array) #1
%3 = tail call spir_func i64 @_Z20get_image_array_size26ocl_image2d_array_depth_ro(ptr addrspace(1) %array) #1
%4 = trunc i64 %3 to i32
%5 = add nsw i32 %2, %1
%6 = add nsw i32 %5, %4
store i32 %5, ptr addrspace(1) %sizes, align 4, !tbaa !22
ret void
}
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z15get_image_width26ocl_image2d_array_depth_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i32 @_Z16get_image_height26ocl_image2d_array_depth_ro(ptr addrspace(1)) #1
; Function Attrs: nounwind readnone
declare spir_func i64 @_Z20get_image_array_size26ocl_image2d_array_depth_ro(ptr addrspace(1)) #1
attributes #0 = { nounwind }
attributes #1 = { nounwind readnone }
!opencl.enable.FP_CONTRACT = !{}
!opencl.spir.version = !{!18}
!opencl.ocl.version = !{!18}
!opencl.used.extensions = !{!19}
!opencl.used.optional.core.features = !{!20}
!opencl.compiler.options = !{!21}
!1 = !{i32 1, i32 1, i32 1, i32 1}
!2 = !{!"none", !"read_only", !"read_only", !"read_only"}
!3 = !{!"int*", !"image1d_t", !"image1d_buffer_t", !"image1d_array_t"}
!4 = !{!"", !"", !"", !""}
!5 = !{!"int*", !"image1d_t", !"image1d_buffer_t", !"image1d_array_t"}
!7 = !{i32 1, i32 1, i32 1, i32 1, i32 1}
!8 = !{!"none", !"read_only", !"read_only", !"read_only", !"read_only"}
!9 = !{!"int*", !"image2d_t", !"image2d_depth_t", !"image2d_array_t", !"image2d_array_depth_t"}
!10 = !{!"", !"", !"", !"", !""}
!11 = !{!"int*", !"image2d_t", !"image2d_depth_t", !"image2d_array_t", !"image2d_array_depth_t"}
!13 = !{i32 1, i32 1}
!14 = !{!"none", !"read_only"}
!15 = !{!"int*", !"image3d_t"}
!16 = !{!"", !""}
!17 = !{!"int*", !"image3d_t"}
!18 = !{i32 1, i32 2}
!19 = !{!"cl_khr_depth_images"}
!20 = !{!"cl_images"}
!21 = !{}
!22 = !{!23, !23, i64 0}
!23 = !{!"int", !24}
!24 = !{!"omnipotent char", !25}
!25 = !{!"Simple C/C++ TBAA"}
!27 = !{i32 1, i32 1}
!28 = !{!"none", !"read_only"}
!29 = !{!"int*", !"image2d_array_depth_t"}
!30 = !{!"", !""}
!31 = !{!"int*", !"image2d_array_depth_t"}