Files
TheRock/compiler/spirv-llvm-translator/test/transcoding/pipe_builtins.cl
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

195 lines
8.5 KiB
Common Lisp

// Pipe built-ins are mangled accordingly to SPIR2.0/C++ ABI.
// RUN: %clang_cc1 -x cl -cl-std=CL2.0 -triple spir64-unknown-unknown -emit-llvm-bc -fdeclare-opencl-builtins -finclude-default-header -Dcl_khr_subgroups %s -o %t.bc
// RUN: llvm-spirv %t.bc -spirv-text -o %t.spt
// RUN: FileCheck < %t.spt %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 --check-prefix=CHECK-LLVM
// CHECK-SPIRV-DAG: EntryPoint [[#]] [[#ConvenienceWriteUint:]] "test_pipe_convenience_write_uint"
// CHECK-SPIRV-DAG: EntryPoint [[#]] [[#ConvenienceReadUint:]] "test_pipe_convenience_read_uint"
// CHECK-SPIRV-DAG: EntryPoint [[#]] [[#Write:]] "test_pipe_write"
// CHECK-SPIRV-DAG: EntryPoint [[#]] [[#Query:]] "test_pipe_query_functions"
// CHECK-SPIRV-DAG: EntryPoint [[#]] [[#Read:]] "test_pipe_read"
// CHECK-SPIRV-DAG: EntryPoint [[#]] [[#WorkgroupWriteChar:]] "test_pipe_workgroup_write_char"
// CHECK-SPIRV-DAG: EntryPoint [[#]] [[#WorkgroupReadChar:]] "test_pipe_workgroup_read_char"
// CHECK-SPIRV-DAG: EntryPoint [[#]] [[#SubgroupWriteUint:]] "test_pipe_subgroup_write_uint"
// CHECK-SPIRV-DAG: EntryPoint [[#]] [[#SubgroupReadUint:]] "test_pipe_subgroup_read_uint"
// CHECK-SPIRV-DAG: TypePipe [[ROPipeTy:[0-9]+]] 0
// CHECK-SPIRV-DAG: TypePipe [[WOPipeTy:[0-9]+]] 1
#pragma OPENCL EXTENSION cl_khr_subgroups : enable
// CHECK-LLVM-LABEL: @test_pipe_convenience_write_uint
__kernel void test_pipe_convenience_write_uint(__global uint *src, __write_only pipe uint out_pipe)
{
// CHECK-SPIRV: Function [[#]] [[#ConvenienceWriteUint]]
// CHECK-SPIRV-NEXT: FunctionParameter
// CHECK-SPIRV-NEXT: FunctionParameter [[WOPipeTy]] [[PipeArgID:[0-9]+]]
// CHECK-LLVM: call{{.*}}@__write_pipe_2
// CHECK-SPIRV: WritePipe {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
int gid = get_global_id(0);
write_pipe(out_pipe, &src[gid]);
}
// CHECK-LLVM-LABEL: @test_pipe_convenience_read_uint
// CHECK-SPIRV: Function [[#]] [[#ConvenienceReadUint]]
__kernel void test_pipe_convenience_read_uint(__read_only pipe uint in_pipe, __global uint *dst)
{
// CHECK-SPIRV-NEXT: FunctionParameter [[ROPipeTy]] [[PipeArgID:[0-9]+]]
// CHECK-LLVM: call{{.*}}@__read_pipe_2
// CHECK-SPIRV: ReadPipe {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
int gid = get_global_id(0);
read_pipe(in_pipe, &dst[gid]);
}
// CHECK-LLVM-LABEL: @test_pipe_write
// CHECK-SPIRV: Function [[#]] [[#Write]]
__kernel void test_pipe_write(__global int *src, __write_only pipe int out_pipe)
{
// CHECK-SPIRV-NEXT: FunctionParameter
// CHECK-SPIRV-NEXT: FunctionParameter [[WOPipeTy]] [[PipeArgID:[0-9]+]]
// CHECK-LLVM: @__reserve_write_pipe
// CHECK-SPIRV: ReserveWritePipePackets {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
// CHECK-LLVM: call{{.*}}@__write_pipe_4
// CHECK-SPIRV: ReservedWritePipe {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
// CHECK-LLVM: call{{.*}}@__commit_write_pipe
// CHECK-SPIRV: CommitWritePipe [[PipeArgID]] {{[0-9]+}} {{[0-9]+}}
int gid = get_global_id(0);
reserve_id_t res_id;
res_id = reserve_write_pipe(out_pipe, 1);
if(is_valid_reserve_id(res_id))
{
write_pipe(out_pipe, res_id, 0, &src[gid]);
commit_write_pipe(out_pipe, res_id);
}
}
// CHECK-LLVM-LABEL: @test_pipe_query_functions
// CHECK-SPIRV: Function [[#]] [[#Query]]
__kernel void test_pipe_query_functions(__write_only pipe int out_pipe, __global int *num_packets, __global int *max_packets)
{
// CHECK-SPIRV-NEXT: FunctionParameter [[WOPipeTy]] [[PipeArgID:[0-9]+]]
// CHECK-LLVM: call{{.*}}@__get_pipe_max_packets_wo
// CHECK-SPIRV: GetMaxPipePackets {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}}
// CHECK-LLVM: call{{.*}}@__get_pipe_num_packets
// CHECK-SPIRV: GetNumPipePackets {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}}
*max_packets = get_pipe_max_packets(out_pipe);
*num_packets = get_pipe_num_packets(out_pipe);
}
// CHECK-LLVM-LABEL: @test_pipe_read
// CHECK-SPIRV: Function [[#]] [[#Read]]
__kernel void test_pipe_read(__read_only pipe int in_pipe, __global int *dst)
{
// CHECK-SPIRV-NEXT: FunctionParameter [[ROPipeTy]] [[PipeArgID:[0-9]+]]
// CHECK-LLVM: call{{.*}}@__reserve_read_pipe
// CHECK-SPIRV: ReserveReadPipePackets {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
// CHECK-LLVM: call{{.*}}@__read_pipe_4
// CHECK-SPIRV: ReservedReadPipe {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
// CHECK-LLVM: call{{.*}}@__commit_read_pipe
// CHECK-SPIRV: CommitReadPipe [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
int gid = get_global_id(0);
reserve_id_t res_id;
res_id = reserve_read_pipe(in_pipe, 1);
if(is_valid_reserve_id(res_id))
{
read_pipe(in_pipe, res_id, 0, &dst[gid]);
commit_read_pipe(in_pipe, res_id);
}
}
// CHECK-LLVM-LABEL: @test_pipe_workgroup_write_char
// CHECK-SPIRV: Function [[#]] [[#WorkgroupWriteChar]]
__kernel void test_pipe_workgroup_write_char(__global char *src, __write_only pipe char out_pipe)
{
// CHECK-SPIRV-NEXT: FunctionParameter
// CHECK-SPIRV-NEXT: FunctionParameter [[WOPipeTy]] [[PipeArgID:[0-9]+]]
// CHECK-LLVM: call{{.*}}@__work_group_reserve_write_pipe
// CHECK-SPIRV: GroupReserveWritePipePackets {{[0-9]+}} {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
// CHECK-LLVM: call{{.*}}@__work_group_commit_write_pipe
// CHECK-SPIRV: GroupCommitWritePipe {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
int gid = get_global_id(0);
__local reserve_id_t res_id;
res_id = work_group_reserve_write_pipe(out_pipe, get_local_size(0));
if(is_valid_reserve_id(res_id))
{
write_pipe(out_pipe, res_id, get_local_id(0), &src[gid]);
work_group_commit_write_pipe(out_pipe, res_id);
}
}
// CHECK-LLVM-LABEL: @test_pipe_workgroup_read_char
// CHECK-SPIRV: Function [[#]] [[#WorkgroupReadChar]]
__kernel void test_pipe_workgroup_read_char(__read_only pipe char in_pipe, __global char *dst)
{
// CHECK-SPIRV-NEXT: FunctionParameter [[ROPipeTy]] [[PipeArgID:[0-9]+]]
// CHECK-LLVM: call{{.*}}@__work_group_reserve_read_pipe
// CHECK-SPIRV: GroupReserveReadPipePackets {{[0-9]+}} {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
// CHECK-LLVM: call{{.*}}@__work_group_commit_read_pipe
// CHECK-SPIRV: GroupCommitReadPipe {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
int gid = get_global_id(0);
__local reserve_id_t res_id;
res_id = work_group_reserve_read_pipe(in_pipe, get_local_size(0));
if(is_valid_reserve_id(res_id))
{
read_pipe(in_pipe, res_id, get_local_id(0), &dst[gid]);
work_group_commit_read_pipe(in_pipe, res_id);
}
}
// CHECK-LLVM-LABEL: @test_pipe_subgroup_write_uint
// CHECK-SPIRV: Function [[#]] [[#SubgroupWriteUint]]
__kernel void test_pipe_subgroup_write_uint(__global uint *src, __write_only pipe uint out_pipe)
{
// CHECK-SPIRV-NEXT: FunctionParameter
// CHECK-SPIRV-NEXT: FunctionParameter [[WOPipeTy]] [[PipeArgID:[0-9]+]]
// CHECK-LLVM: call{{.*}}@__sub_group_reserve_write_pipe
// CHECK-SPIRV: GroupReserveWritePipePackets {{[0-9]+}} {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
// CHECK-LLVM: call{{.*}}@__sub_group_commit_write_pipe
// CHECK-SPIRV: GroupCommitWritePipe {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
int gid = get_global_id(0);
reserve_id_t res_id;
res_id = sub_group_reserve_write_pipe(out_pipe, get_sub_group_size());
if(is_valid_reserve_id(res_id))
{
write_pipe(out_pipe, res_id, get_sub_group_local_id(), &src[gid]);
sub_group_commit_write_pipe(out_pipe, res_id);
}
}
// CHECK-LLVM-LABEL: @test_pipe_subgroup_read_uint
// CHECK-SPIRV: Function [[#]] [[#SubgroupReadUint]]
__kernel void test_pipe_subgroup_read_uint(__read_only pipe uint in_pipe, __global uint *dst)
{
// CHECK-SPIRV-NEXT: FunctionParameter [[ROPipeTy]] [[PipeArgID:[0-9]+]]
// CHECK-LLVM: call{{.*}}@__sub_group_reserve_read_pipe
// CHECK-SPIRV: GroupReserveReadPipePackets {{[0-9]+}} {{[0-9]+}} {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
// CHECK-LLVM: call{{.*}}@__sub_group_commit_read_pipe
// CHECK-SPIRV: GroupCommitReadPipe {{[0-9]+}} [[PipeArgID]] {{[0-9]+}} {{[0-9]+}} {{[0-9]+}}
int gid = get_global_id(0);
reserve_id_t res_id;
res_id = sub_group_reserve_read_pipe(in_pipe, get_sub_group_size());
if(is_valid_reserve_id(res_id))
{
read_pipe(in_pipe, res_id, get_sub_group_local_id(), &dst[gid]);
sub_group_commit_read_pipe(in_pipe, res_id);
}
}