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
161 lines
6.9 KiB
MLIR
161 lines
6.9 KiB
MLIR
// RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip %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 @math(%arg0 : f32, %arg1 : f32, %arg2 : i32) "None" {
|
|
// CHECK: {{%.*}} = spirv.GL.Exp {{%.*}} : f32
|
|
%0 = spirv.GL.Exp %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Sqrt {{%.*}} : f32
|
|
%2 = spirv.GL.Sqrt %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Cos {{%.*}} : f32
|
|
%3 = spirv.GL.Cos %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Sin {{%.*}} : f32
|
|
%4 = spirv.GL.Sin %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Tan {{%.*}} : f32
|
|
%5 = spirv.GL.Tan %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Acos {{%.*}} : f32
|
|
%6 = spirv.GL.Acos %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Asin {{%.*}} : f32
|
|
%7 = spirv.GL.Asin %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Atan {{%.*}} : f32
|
|
%8 = spirv.GL.Atan %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Sinh {{%.*}} : f32
|
|
%9 = spirv.GL.Sinh %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Cosh {{%.*}} : f32
|
|
%10 = spirv.GL.Cosh %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Pow {{%.*}} : f32
|
|
%11 = spirv.GL.Pow %arg0, %arg1 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Round {{%.*}} : f32
|
|
%12 = spirv.GL.Round %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Trunc {{%.*}} : f32
|
|
%trunc = spirv.GL.Trunc %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.FrexpStruct {{%.*}} : f32 -> !spirv.struct<(f32, i32)>
|
|
%13 = spirv.GL.FrexpStruct %arg0 : f32 -> !spirv.struct<(f32, i32)>
|
|
// CHECK: {{%.*}} = spirv.GL.Ldexp {{%.*}} : f32, {{%.*}} : i32 -> f32
|
|
%14 = spirv.GL.Ldexp %arg0 : f32, %arg2 : i32 -> f32
|
|
// CHECK: {{%.*}} = spirv.GL.FMix {{%.*}} : f32, {{%.*}} : f32, {{%.*}} : f32 -> f32
|
|
%15 = spirv.GL.FMix %arg0 : f32, %arg1 : f32, %arg0 : f32 -> f32
|
|
// CHECK: {{%.*}} = spirv.GL.Fract {{%.*}} : f32
|
|
%16 = spirv.GL.Fract %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Asinh {{%.*}} : f32
|
|
%17 = spirv.GL.Asinh %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Acosh {{%.*}} : f32
|
|
%18 = spirv.GL.Acosh %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Atanh {{%.*}} : f32
|
|
%19 = spirv.GL.Atanh %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Log2 {{%.*}} : f32
|
|
%20 = spirv.GL.Log2 %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Tanh {{%.*}} : f32
|
|
%21 = spirv.GL.Tanh %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Exp2 {{%.*}} : f32
|
|
%22 = spirv.GL.Exp2 %arg0 : f32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @maxmin(%arg0 : f32, %arg1 : f32, %arg2 : i32, %arg3 : i32) "None" {
|
|
// CHECK: {{%.*}} = spirv.GL.FMax {{%.*}}, {{%.*}} : f32
|
|
%1 = spirv.GL.FMax %arg0, %arg1 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.SMax {{%.*}}, {{%.*}} : i32
|
|
%2 = spirv.GL.SMax %arg2, %arg3 : i32
|
|
// CHECK: {{%.*}} = spirv.GL.UMax {{%.*}}, {{%.*}} : i32
|
|
%3 = spirv.GL.UMax %arg2, %arg3 : i32
|
|
|
|
// CHECK: {{%.*}} = spirv.GL.FMin {{%.*}}, {{%.*}} : f32
|
|
%4 = spirv.GL.FMin %arg0, %arg1 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.SMin {{%.*}}, {{%.*}} : i32
|
|
%5 = spirv.GL.SMin %arg2, %arg3 : i32
|
|
// CHECK: {{%.*}} = spirv.GL.UMin {{%.*}}, {{%.*}} : i32
|
|
%6 = spirv.GL.UMin %arg2, %arg3 : i32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @fclamp(%arg0 : f32, %arg1 : f32, %arg2 : f32) "None" {
|
|
// CHECK: spirv.GL.FClamp {{%[^,]*}}, {{%[^,]*}}, {{%[^,]*}} : f32
|
|
%13 = spirv.GL.FClamp %arg0, %arg1, %arg2 : f32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @uclamp(%arg0 : ui32, %arg1 : ui32, %arg2 : ui32) "None" {
|
|
// CHECK: spirv.GL.UClamp {{%[^,]*}}, {{%[^,]*}}, {{%[^,]*}} : i32
|
|
%13 = spirv.GL.UClamp %arg0, %arg1, %arg2 : ui32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @sclamp(%arg0 : si32, %arg1 : si32, %arg2 : si32) "None" {
|
|
// CHECK: spirv.GL.SClamp {{%[^,]*}}, {{%[^,]*}}, {{%[^,]*}} : si32
|
|
%13 = spirv.GL.SClamp %arg0, %arg1, %arg2 : si32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @fma(%arg0 : f32, %arg1 : f32, %arg2 : f32) "None" {
|
|
// CHECK: spirv.GL.Fma {{%[^,]*}}, {{%[^,]*}}, {{%[^,]*}} : f32
|
|
%13 = spirv.GL.Fma %arg0, %arg1, %arg2 : f32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @findilsb(%arg0 : i32) "None" {
|
|
// CHECK: spirv.GL.FindILsb {{%.*}} : i32
|
|
%2 = spirv.GL.FindILsb %arg0 : i32
|
|
spirv.Return
|
|
}
|
|
spirv.func @findsmsb(%arg0 : i32) "None" {
|
|
// CHECK: spirv.GL.FindSMsb {{%.*}} : i32
|
|
%2 = spirv.GL.FindSMsb %arg0 : i32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @findumsb(%arg0 : i32) "None" {
|
|
// CHECK: spirv.GL.FindUMsb {{%.*}} : i32
|
|
%2 = spirv.GL.FindUMsb %arg0 : i32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @vector(%arg0 : f32, %arg1 : vector<3xf32>, %arg2 : vector<3xf32>, %arg3: vector<3xi32>) "None" {
|
|
// CHECK: {{%.*}} = spirv.GL.Cross {{%.*}}, {{%.*}} : vector<3xf32>
|
|
%0 = spirv.GL.Cross %arg1, %arg2 : vector<3xf32>
|
|
// CHECK: {{%.*}} = spirv.GL.Normalize {{%.*}} : f32
|
|
%1 = spirv.GL.Normalize %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Normalize {{%.*}} : vector<3xf32>
|
|
%2 = spirv.GL.Normalize %arg1 : vector<3xf32>
|
|
// CHECK: {{%.*}} = spirv.GL.Reflect {{%.*}}, {{%.*}} : f32
|
|
%3 = spirv.GL.Reflect %arg0, %arg0 : f32
|
|
// CHECK: {{%.*}} = spirv.GL.Reflect {{%.*}}, {{%.*}} : vector<3xf32>
|
|
%4 = spirv.GL.Reflect %arg1, %arg2 : vector<3xf32>
|
|
// CHECK: {{%.*}} = spirv.GL.Distance {{%.*}}, {{%.*}} : f32, f32 -> f32
|
|
%5 = spirv.GL.Distance %arg0, %arg0 : f32, f32 -> f32
|
|
// CHECK: {{%.*}} = spirv.GL.Distance {{%.*}}, {{%.*}} : vector<3xf32>, vector<3xf32> -> f32
|
|
%6 = spirv.GL.Distance %arg1, %arg2 : vector<3xf32>, vector<3xf32> -> f32
|
|
// CHECK: {{%.*}} = spirv.GL.FindILsb {{%.*}} : vector<3xi32>
|
|
%7 = spirv.GL.FindILsb %arg3 : vector<3xi32>
|
|
// CHECK: {{%.*}} = spirv.GL.FindSMsb {{%.*}} : vector<3xi32>
|
|
%8 = spirv.GL.FindSMsb %arg3 : vector<3xi32>
|
|
// CHECK: {{%.*}} = spirv.GL.FindUMsb {{%.*}} : vector<3xi32>
|
|
%9 = spirv.GL.FindUMsb %arg3 : vector<3xi32>
|
|
// CHECK: {{%.*}} = spirv.GL.Length {{%.*}} : f32 -> f32
|
|
%10 = spirv.GL.Length %arg0 : f32 -> f32
|
|
// CHECK: {{%.*}} = spirv.GL.Length {{%.*}} : vector<3xf32> -> f32
|
|
%11 = spirv.GL.Length %arg1 : vector<3xf32> -> f32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @pack_half_2x16(%arg0 : i32) "None" {
|
|
// CHECK: {{%.*}} = spirv.GL.UnpackHalf2x16 {{%.*}} : i32 -> vector<2xf32>
|
|
%0 = spirv.GL.UnpackHalf2x16 %arg0 : i32 -> vector<2xf32>
|
|
// CHECK: {{%.*}} = spirv.GL.PackHalf2x16 {{%.*}} : vector<2xf32> -> i32
|
|
%1 = spirv.GL.PackHalf2x16 %0 : vector<2xf32> -> i32
|
|
spirv.Return
|
|
}
|
|
|
|
spirv.func @pack_snorm_4x8(%arg0 : i32) "None" {
|
|
// CHECK: {{%.*}} = spirv.GL.UnpackSnorm4x8 {{%.*}} : i32 -> vector<4xf32>
|
|
%0 = spirv.GL.UnpackSnorm4x8 %arg0 : i32 -> vector<4xf32>
|
|
// CHECK: {{%.*}} = spirv.GL.PackSnorm4x8 {{%.*}} : vector<4xf32> -> i32
|
|
%1 = spirv.GL.PackSnorm4x8 %0 : vector<4xf32> -> i32
|
|
spirv.Return
|
|
}
|
|
}
|