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
199 lines
8.7 KiB
MLIR
199 lines
8.7 KiB
MLIR
// RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip -split-input-file %s | FileCheck %s
|
|
|
|
// RUN: %if spirv-tools %{ rm -rf %t %}
|
|
// RUN: %if spirv-tools %{ mkdir %t %}
|
|
// RUN: %if spirv-tools %{ mlir-translate --no-implicit-module --serialize-spirv --split-input-file --spirv-save-validation-files-with-prefix=%t/module %s %}
|
|
// RUN: %if spirv-tools %{ spirv-val %t %}
|
|
|
|
spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader, Linkage], []> {
|
|
spirv.func @bit_cast(%arg0 : f32) "None" {
|
|
// CHECK: {{%.*}} = spirv.Bitcast {{%.*}} : f32 to i32
|
|
%0 = spirv.Bitcast %arg0 : f32 to i32
|
|
// CHECK: {{%.*}} = spirv.Bitcast {{%.*}} : i32 to si32
|
|
%1 = spirv.Bitcast %0 : i32 to si32
|
|
// CHECK: {{%.*}} = spirv.Bitcast {{%.*}} : si32 to i32
|
|
%2 = spirv.Bitcast %1 : si32 to ui32
|
|
spirv.Return
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
spirv.module Logical Vulkan requires #spirv.vce<v1.6, [Shader, Linkage, CooperativeMatrixKHR, VulkanMemoryModel], [SPV_KHR_cooperative_matrix, SPV_KHR_vulkan_memory_model]> {
|
|
spirv.func @bit_cast_coop(%arg0 : !spirv.coopmatrix<4x4xf32, Subgroup, MatrixA>) "None" {
|
|
// CHECK: {{%.*}} = spirv.Bitcast {{%.*}} : !spirv.coopmatrix<4x4xf32, Subgroup, MatrixA> to !spirv.coopmatrix<4x4xi32, Subgroup, MatrixA>
|
|
%0 = spirv.Bitcast %arg0 : !spirv.coopmatrix<4x4xf32, Subgroup, MatrixA> to !spirv.coopmatrix<4x4xi32, Subgroup, MatrixA>
|
|
spirv.Return
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader, Linkage, BFloat16TypeKHR, Float64, Int64], [SPV_KHR_bfloat16]> {
|
|
spirv.func @convert_f_to_s(%arg0 : f32) -> i32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertFToS {{%.*}} : f32 to i32
|
|
%0 = spirv.ConvertFToS %arg0 : f32 to i32
|
|
spirv.ReturnValue %0 : i32
|
|
}
|
|
spirv.func @convert_f64_to_s32(%arg0 : f64) -> i32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertFToS {{%.*}} : f64 to i32
|
|
%0 = spirv.ConvertFToS %arg0 : f64 to i32
|
|
spirv.ReturnValue %0 : i32
|
|
}
|
|
spirv.func @convert_bf16_to_s32(%arg0 : bf16) -> i32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertFToS {{%.*}} : bf16 to i32
|
|
%0 = spirv.ConvertFToS %arg0 : bf16 to i32
|
|
spirv.ReturnValue %0 : i32
|
|
}
|
|
spirv.func @convert_f_to_u(%arg0 : f32) -> i32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertFToU {{%.*}} : f32 to i32
|
|
%0 = spirv.ConvertFToU %arg0 : f32 to i32
|
|
spirv.ReturnValue %0 : i32
|
|
}
|
|
spirv.func @convert_f64_to_u32(%arg0 : f64) -> i32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertFToU {{%.*}} : f64 to i32
|
|
%0 = spirv.ConvertFToU %arg0 : f64 to i32
|
|
spirv.ReturnValue %0 : i32
|
|
}
|
|
spirv.func @convert_bf16_to_u32(%arg0 : bf16) -> i32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertFToU {{%.*}} : bf16 to i32
|
|
%0 = spirv.ConvertFToU %arg0 : bf16 to i32
|
|
spirv.ReturnValue %0 : i32
|
|
}
|
|
spirv.func @convert_s_to_f(%arg0 : i32) -> f32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertSToF {{%.*}} : i32 to f32
|
|
%0 = spirv.ConvertSToF %arg0 : i32 to f32
|
|
spirv.ReturnValue %0 : f32
|
|
}
|
|
spirv.func @convert_s64_to_f32(%arg0 : i64) -> f32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertSToF {{%.*}} : i64 to f32
|
|
%0 = spirv.ConvertSToF %arg0 : i64 to f32
|
|
spirv.ReturnValue %0 : f32
|
|
}
|
|
spirv.func @convert_s64_to_bf16(%arg0 : i64) -> bf16 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertSToF {{%.*}} : i64 to bf16
|
|
%0 = spirv.ConvertSToF %arg0 : i64 to bf16
|
|
spirv.ReturnValue %0 : bf16
|
|
}
|
|
spirv.func @convert_u_to_f(%arg0 : i32) -> f32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertUToF {{%.*}} : i32 to f32
|
|
%0 = spirv.ConvertUToF %arg0 : i32 to f32
|
|
spirv.ReturnValue %0 : f32
|
|
}
|
|
spirv.func @convert_u64_to_f32(%arg0 : i64) -> f32 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertUToF {{%.*}} : i64 to f32
|
|
%0 = spirv.ConvertUToF %arg0 : i64 to f32
|
|
spirv.ReturnValue %0 : f32
|
|
}
|
|
spirv.func @convert_u64_to_bf16(%arg0 : i64) -> bf16 "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertUToF {{%.*}} : i64 to bf16
|
|
%0 = spirv.ConvertUToF %arg0 : i64 to bf16
|
|
spirv.ReturnValue %0 : bf16
|
|
}
|
|
spirv.func @convert_f32_to_f64(%arg0 : f32) -> f64 "None" {
|
|
// CHECK: {{%.*}} = spirv.FConvert {{%.*}} : f32 to f64
|
|
%0 = spirv.FConvert %arg0 : f32 to f64
|
|
spirv.ReturnValue %0 : f64
|
|
}
|
|
spirv.func @convert_f32_to_bf16(%arg0 : f32) -> bf16 "None" {
|
|
// CHECK: {{%.*}} = spirv.FConvert {{%.*}} : f32 to bf16
|
|
%0 = spirv.FConvert %arg0 : f32 to bf16
|
|
spirv.ReturnValue %0 : bf16
|
|
}
|
|
spirv.func @convert_bf16_to_f32(%arg0 : bf16) -> f32 "None" {
|
|
// CHECK: {{%.*}} = spirv.FConvert {{%.*}} : bf16 to f32
|
|
%0 = spirv.FConvert %arg0 : bf16 to f32
|
|
spirv.ReturnValue %0 : f32
|
|
}
|
|
spirv.func @s_convert(%arg0 : i32) -> i64 "None" {
|
|
// CHECK: {{%.*}} = spirv.SConvert {{%.*}} : i32 to i64
|
|
%0 = spirv.SConvert %arg0 : i32 to i64
|
|
spirv.ReturnValue %0 : i64
|
|
}
|
|
spirv.func @u_convert(%arg0 : i32) -> i64 "None" {
|
|
// CHECK: {{%.*}} = spirv.UConvert {{%.*}} : i32 to i64
|
|
%0 = spirv.UConvert %arg0 : i32 to i64
|
|
spirv.ReturnValue %0 : i64
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
spirv.module Physical64 OpenCL requires #spirv.vce<v1.0, [Kernel, Linkage, GenericPointer], []> {
|
|
spirv.func @ptr_cast_to_generic(%arg0 : !spirv.ptr<f32, CrossWorkgroup>) "None" {
|
|
// CHECK: {{%.*}} = spirv.PtrCastToGeneric {{%.*}} : !spirv.ptr<f32, CrossWorkgroup> to !spirv.ptr<f32, Generic>
|
|
%0 = spirv.PtrCastToGeneric %arg0 : !spirv.ptr<f32, CrossWorkgroup> to !spirv.ptr<f32, Generic>
|
|
spirv.Return
|
|
}
|
|
spirv.func @generic_cast_to_ptr(%arg0 : !spirv.ptr<vector<2xi32>, Generic>) "None" {
|
|
// CHECK: {{%.*}} = spirv.GenericCastToPtr {{%.*}} : !spirv.ptr<vector<2xi32>, Generic> to !spirv.ptr<vector<2xi32>, CrossWorkgroup>
|
|
%0 = spirv.GenericCastToPtr %arg0 : !spirv.ptr<vector<2xi32>, Generic> to !spirv.ptr<vector<2xi32>, CrossWorkgroup>
|
|
spirv.Return
|
|
}
|
|
spirv.func @generic_cast_to_ptr_explicit(%arg0 : !spirv.ptr<vector<2xi32>, Generic>) "None" {
|
|
// CHECK: {{%.*}} = spirv.GenericCastToPtrExplicit {{%.*}} : !spirv.ptr<vector<2xi32>, Generic> to !spirv.ptr<vector<2xi32>, CrossWorkgroup>
|
|
%0 = spirv.GenericCastToPtrExplicit %arg0 : !spirv.ptr<vector<2xi32>, Generic> to !spirv.ptr<vector<2xi32>, CrossWorkgroup>
|
|
spirv.Return
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
spirv.module Physical64 OpenCL requires #spirv.vce<v1.0, [Kernel, Addresses, Linkage, GenericPointer, Int64], []> {
|
|
spirv.func @covert_ptr_to_u(%arg0 : !spirv.ptr<i32, Generic>) "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertPtrToU {{%.*}} : !spirv.ptr<i32, Generic> to i32
|
|
%0 = spirv.ConvertPtrToU %arg0 : !spirv.ptr<i32, Generic> to i32
|
|
spirv.Return
|
|
}
|
|
spirv.func @covert_ptr_to_u_truncate(%arg0 : !spirv.ptr<i64, Generic>) "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertPtrToU {{%.*}} : !spirv.ptr<i64, Generic> to i32
|
|
%0 = spirv.ConvertPtrToU %arg0 : !spirv.ptr<i64, Generic> to i32
|
|
spirv.Return
|
|
}
|
|
spirv.func @covert_ptr_to_u_extend(%arg0 : !spirv.ptr<i32, Generic>) "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertPtrToU {{%.*}} : !spirv.ptr<i32, Generic> to i64
|
|
%0 = spirv.ConvertPtrToU %arg0 : !spirv.ptr<i32, Generic> to i64
|
|
spirv.Return
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
spirv.module PhysicalStorageBuffer64 OpenCL requires #spirv.vce<v1.0, [Kernel, Addresses, PhysicalStorageBufferAddresses, Linkage], [SPV_EXT_physical_storage_buffer]> {
|
|
spirv.func @covert_ptr_to_u_PhysicalStorageBuffer(%arg0 : !spirv.ptr<i32, PhysicalStorageBuffer> { spirv.decoration = #spirv.decoration<Aliased>} ) "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertPtrToU {{%.*}} : !spirv.ptr<i32, PhysicalStorageBuffer> to i32
|
|
%0 = spirv.ConvertPtrToU %arg0 : !spirv.ptr<i32, PhysicalStorageBuffer> to i32
|
|
spirv.Return
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
spirv.module Physical64 OpenCL requires #spirv.vce<v1.0, [Kernel, Addresses, Linkage, GenericPointer, Int64], []> {
|
|
spirv.func @covert_u_to_ptr(%arg0 : i32) "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertUToPtr {{%.*}} : i32 to !spirv.ptr<i32, Generic>
|
|
%0 = spirv.ConvertUToPtr %arg0 : i32 to !spirv.ptr<i32, Generic>
|
|
spirv.Return
|
|
}
|
|
spirv.func @covert_u_to_ptr_truncate(%arg0 : i64) "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertUToPtr {{%.*}} : i64 to !spirv.ptr<i32, Generic>
|
|
%0 = spirv.ConvertUToPtr %arg0 : i64 to !spirv.ptr<i32, Generic>
|
|
spirv.Return
|
|
}
|
|
spirv.func @covert_u_to_ptr_extend(%arg0 : i32) "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertUToPtr {{%.*}} : i32 to !spirv.ptr<i64, Generic>
|
|
%0 = spirv.ConvertUToPtr %arg0 : i32 to !spirv.ptr<i64, Generic>
|
|
spirv.Return
|
|
}
|
|
}
|
|
|
|
// -----
|
|
|
|
spirv.module PhysicalStorageBuffer64 OpenCL requires #spirv.vce<v1.0, [Kernel, Addresses, PhysicalStorageBufferAddresses, Linkage], [SPV_EXT_physical_storage_buffer]> {
|
|
spirv.func @covert_u_to_ptr_PhysicalStorageBuffer(%arg0 : i32) "None" {
|
|
// CHECK: {{%.*}} = spirv.ConvertUToPtr {{%.*}} : i32 to !spirv.ptr<i32, PhysicalStorageBuffer>
|
|
%0 = spirv.ConvertUToPtr %arg0 : i32 to !spirv.ptr<i32, PhysicalStorageBuffer>
|
|
spirv.Return
|
|
}
|
|
}
|