113 lines · plain
1; RUN: llc -O0 -mtriple=spirv32-unknown-unknown %s -o - | FileCheck %s --check-prefix=CHECK-SPIRV2; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv32-unknown-unknown %s -o - -filetype=obj | spirv-val %}3 4;; This test checks that the backend is capable to correctly translate5;; sub_group_barrier built-in function [1] from cl_khr_subgroups extension into6;; corresponding SPIR-V instruction.7 8;; __kernel void test_barrier_const_flags() {9;; work_group_barrier(CLK_LOCAL_MEM_FENCE);10;; work_group_barrier(CLK_GLOBAL_MEM_FENCE);11;; work_group_barrier(CLK_IMAGE_MEM_FENCE);12;;13;; work_group_barrier(CLK_LOCAL_MEM_FENCE | CLK_GLOBAL_MEM_FENCE);14;; work_group_barrier(CLK_LOCAL_MEM_FENCE | CLK_IMAGE_MEM_FENCE);15;; work_group_barrier(CLK_GLOBAL_MEM_FENCE | CLK_LOCAL_MEM_FENCE | CLK_IMAGE_MEM_FENCE);16;;17;; work_group_barrier(CLK_LOCAL_MEM_FENCE, memory_scope_work_item);18;; work_group_barrier(CLK_LOCAL_MEM_FENCE, memory_scope_work_group);19;; work_group_barrier(CLK_LOCAL_MEM_FENCE, memory_scope_device);20;; work_group_barrier(CLK_LOCAL_MEM_FENCE, memory_scope_all_svm_devices);21;; work_group_barrier(CLK_LOCAL_MEM_FENCE, memory_scope_sub_group);22;;23 ;; barrier should also work (preserved for backward compatibility)24;; barrier(CLK_GLOBAL_MEM_FENCE);25;; }26;;27;; __kernel void test_barrier_non_const_flags(cl_mem_fence_flags flags, memory_scope scope) {28 ;; FIXME: OpenCL spec doesn't require flags to be compile-time known29 ;; work_group_barrier(flags);30 ;; work_group_barrier(flags, scope);31;; }32 33; CHECK-SPIRV: OpName %[[#TEST_CONST_FLAGS:]] "test_barrier_const_flags"34; CHECK-SPIRV: %[[#UINT:]] = OpTypeInt 32 035 36;; In SPIR-V, barrier is represented as OpControlBarrier [2] and OpenCL37;; cl_mem_fence_flags are represented as part of Memory Semantics [3], which38;; also includes memory order constraints. The backend applies some default39;; memory order for OpControlBarrier and therefore, constants below include a40;; bit more information than original source41 42;; 0x10 SequentiallyConsistent + 0x100 WorkgroupMemory43; CHECK-SPIRV-DAG: %[[#LOCAL:]] = OpConstant %[[#UINT]] 27244;; 0x10 SequentiallyConsistent + 0x200 CrossWorkgroupMemory45; CHECK-SPIRV-DAG: %[[#GLOBAL:]] = OpConstant %[[#UINT]] 52846;; 0x10 SequentiallyConsistent + 0x800 ImageMemory47; CHECK-SPIRV-DAG: %[[#IMAGE:]] = OpConstant %[[#UINT]] 206448;; 0x10 SequentiallyConsistent + 0x100 WorkgroupMemory + 0x200 CrossWorkgroupMemory49; CHECK-SPIRV-DAG: %[[#LOCAL_GLOBAL:]] = OpConstant %[[#UINT]] 78450;; 0x10 SequentiallyConsistent + 0x100 WorkgroupMemory + 0x800 ImageMemory51; CHECK-SPIRV-DAG: %[[#LOCAL_IMAGE:]] = OpConstant %[[#UINT]] 232052;; 0x10 SequentiallyConsistent + 0x100 WorkgroupMemory + 0x200 CrossWorkgroupMemory + 0x800 ImageMemory53; CHECK-SPIRV-DAG: %[[#LOCAL_GLOBAL_IMAGE:]] = OpConstant %[[#UINT]] 283254 55;; Scopes [4]:56;; 2 Workgroup57; CHECK-SPIRV-DAG: %[[#SCOPE_WORK_GROUP:]] = OpConstant %[[#UINT]] 2{{$}}58;; 4 Invocation59; CHECK-SPIRV-DAG: %[[#SCOPE_INVOCATION:]] = OpConstant %[[#UINT]] 4{{$}}60;; 1 Device61; CHECK-SPIRV-DAG: %[[#SCOPE_DEVICE:]] = OpConstant %[[#UINT]] 1{{$}}62;; 0 CrossDevice63; CHECK-SPIRV-DAG: %[[#SCOPE_CROSS_DEVICE:]] = OpConstantNull %[[#UINT]]64;; 3 Subgroup65; CHECK-SPIRV-DAG: %[[#SCOPE_SUBGROUP:]] = OpConstant %[[#UINT]] 3{{$}}66 67; CHECK-SPIRV: %[[#TEST_CONST_FLAGS]] = OpFunction %[[#]]68; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_WORK_GROUP]] %[[#LOCAL]]69; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_WORK_GROUP]] %[[#GLOBAL]]70; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_WORK_GROUP]] %[[#IMAGE]]71; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_WORK_GROUP]] %[[#LOCAL_GLOBAL]]72; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_WORK_GROUP]] %[[#LOCAL_IMAGE]]73; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_WORK_GROUP]] %[[#LOCAL_GLOBAL_IMAGE]]74; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_INVOCATION]] %[[#LOCAL]]75; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_WORK_GROUP]] %[[#LOCAL]]76; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_DEVICE]] %[[#LOCAL]]77; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_CROSS_DEVICE]] %[[#LOCAL]]78; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_SUBGROUP]] %[[#LOCAL]]79; CHECK-SPIRV: OpControlBarrier %[[#SCOPE_WORK_GROUP]] %[[#SCOPE_WORK_GROUP]] %[[#GLOBAL]]80 81define dso_local spir_kernel void @test_barrier_const_flags() local_unnamed_addr {82entry:83 tail call spir_func void @_Z18work_group_barrierj(i32 noundef 1)84 tail call spir_func void @_Z18work_group_barrierj(i32 noundef 2)85 tail call spir_func void @_Z18work_group_barrierj(i32 noundef 4)86 tail call spir_func void @_Z18work_group_barrierj(i32 noundef 3)87 tail call spir_func void @_Z18work_group_barrierj(i32 noundef 5)88 tail call spir_func void @_Z18work_group_barrierj(i32 noundef 7)89 tail call spir_func void @_Z18work_group_barrierj12memory_scope(i32 noundef 1, i32 noundef 0)90 tail call spir_func void @_Z18work_group_barrierj12memory_scope(i32 noundef 1, i32 noundef 1)91 tail call spir_func void @_Z18work_group_barrierj12memory_scope(i32 noundef 1, i32 noundef 2)92 tail call spir_func void @_Z18work_group_barrierj12memory_scope(i32 noundef 1, i32 noundef 3)93 tail call spir_func void @_Z18work_group_barrierj12memory_scope(i32 noundef 1, i32 noundef 4)94 tail call spir_func void @_Z7barrierj(i32 noundef 2)95 ret void96}97 98declare spir_func void @_Z18work_group_barrierj(i32 noundef) local_unnamed_addr99 100declare spir_func void @_Z18work_group_barrierj12memory_scope(i32 noundef, i32 noundef) local_unnamed_addr101 102declare spir_func void @_Z7barrierj(i32 noundef) local_unnamed_addr103 104define dso_local spir_kernel void @test_barrier_non_const_flags(i32 noundef %flags, i32 noundef %scope) local_unnamed_addr {105entry:106 ret void107}108 109;; References:110;; [1]: https://www.khronos.org/registry/OpenCL/sdk/2.0/docs/man/xhtml/work_group_barrier.html111;; [2]: https://www.khronos.org/registry/spir-v/specs/unified1/SPIRV.html#OpControlBarrier112;; [3]: https://www.khronos.org/registry/spir-v/specs/unified1/SPIRV.html#_a_id_memory_semantics__id_a_memory_semantics_lt_id_gt113