derek/TheRock
1
0 Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity
Files
1a182e2dac58879f215c0fad0f5f78c76dcafe56
TheRock/compiler/amd-llvm/mlir/test/Conversion/MathToSPIRV/math-to-opencl-spirv.mlir
T
derek.holloway 7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
Details
CodeQL / Analyze (python) (push) Canceled after 0s
Details
pre-commit / pre-commit (push) Canceled after 0s
Details
Multi-Arch CI / setup (push) Canceled after 0s
Details
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Details
Setup (Multi-Arch) / setup (push) Canceled after 0s
Details
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Details
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
Details
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Details
Multi-Arch CI / CI Summary (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Details
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Details
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Details
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Details
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Details
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
Details
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

241 lines
8.4 KiB
MLIR
Raw Blame History

// RUN: mlir-opt -split-input-file -convert-math-to-spirv -verify-diagnostics %s -o - | FileCheck %s
module attributes { spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [Kernel], []>, #spirv.resource_limits<>> } {
// CHECK-LABEL: @float32_unary_scalar
func.func @float32_unary_scalar(%arg0: f32) {
// CHECK: spirv.CL.atan %{{.*}}: f32
%0 = math.atan %arg0 : f32
// CHECK: spirv.CL.cos %{{.*}}: f32
%1 = math.cos %arg0 : f32
// CHECK: spirv.CL.exp %{{.*}}: f32
%2 = math.exp %arg0 : f32
// CHECK: %[[EXP:.+]] = spirv.CL.exp %arg0
// CHECK: %[[ONE:.+]] = spirv.Constant 1.000000e+00 : f32
// CHECK: spirv.FSub %[[EXP]], %[[ONE]]
%3 = math.expm1 %arg0 : f32
// CHECK: spirv.CL.log %{{.*}}: f32
%4 = math.log %arg0 : f32
// CHECK: %[[ONE:.+]] = spirv.Constant 1.000000e+00 : f32
// CHECK: %[[ADDONE:.+]] = spirv.FAdd %[[ONE]], %{{.+}}
// CHECK: spirv.CL.log %[[ADDONE]]
%5 = math.log1p %arg0 : f32
// CHECK: spirv.CL.log2 %{{.*}}: f32
%6 = math.log2 %arg0 : f32
// CHECK: spirv.CL.log10 %{{.*}}: f32
%7 = math.log10 %arg0 : f32
// CHECK: spirv.CL.exp2 %{{.*}}: f32
%exp2_scalar = math.exp2 %arg0 : f32
// CHECK: spirv.CL.rint %{{.*}}: f32
%8 = math.roundeven %arg0 : f32
// CHECK: spirv.CL.rsqrt %{{.*}}: f32
%9 = math.rsqrt %arg0 : f32
// CHECK: spirv.CL.sqrt %{{.*}}: f32
%10 = math.sqrt %arg0 : f32
// CHECK: spirv.CL.tanh %{{.*}}: f32
%11 = math.tanh %arg0 : f32
// CHECK: spirv.CL.sin %{{.*}}: f32
%12 = math.sin %arg0 : f32
// CHECK: spirv.CL.fabs %{{.*}}: f32
%13 = math.absf %arg0 : f32
// CHECK: spirv.CL.ceil %{{.*}}: f32
%14 = math.ceil %arg0 : f32
// CHECK: spirv.CL.floor %{{.*}}: f32
%15 = math.floor %arg0 : f32
// CHECK: spirv.CL.erf %{{.*}}: f32
%16 = math.erf %arg0 : f32
// CHECK: spirv.CL.round %{{.*}}: f32
%17 = math.round %arg0 : f32
// CHECK: spirv.CL.tan %{{.*}}: f32
%18 = math.tan %arg0 : f32
// CHECK: spirv.CL.asin %{{.*}}: f32
%19 = math.asin %arg0 : f32
// CHECK: spirv.CL.acos %{{.*}}: f32
%20 = math.acos %arg0 : f32
// CHECK: spirv.CL.sinh %{{.*}}: f32
%21 = math.sinh %arg0 : f32
// CHECK: spirv.CL.cosh %{{.*}}: f32
%22 = math.cosh %arg0 : f32
// CHECK: spirv.CL.asinh %{{.*}}: f32
%23 = math.asinh %arg0 : f32
// CHECK: spirv.CL.acosh %{{.*}}: f32
%24 = math.acosh %arg0 : f32
// CHECK: spirv.CL.atanh %{{.*}}: f32
%25 = math.atanh %arg0 : f32
// CHECK: spirv.CL.trunc %{{.*}}: f32
%26 = math.trunc %arg0 : f32
return
}
// CHECK-LABEL: @float32_unary_vector
func.func @float32_unary_vector(%arg0: vector<3xf32>) {
// CHECK: spirv.CL.atan %{{.*}}: vector<3xf32>
%0 = math.atan %arg0 : vector<3xf32>
// CHECK: spirv.CL.cos %{{.*}}: vector<3xf32>
%1 = math.cos %arg0 : vector<3xf32>
// CHECK: spirv.CL.exp %{{.*}}: vector<3xf32>
%2 = math.exp %arg0 : vector<3xf32>
// CHECK: %[[EXP:.+]] = spirv.CL.exp %arg0
// CHECK: %[[ONE:.+]] = spirv.Constant dense<1.000000e+00> : vector<3xf32>
// CHECK: spirv.FSub %[[EXP]], %[[ONE]]
%3 = math.expm1 %arg0 : vector<3xf32>
// CHECK: spirv.CL.log %{{.*}}: vector<3xf32>
%4 = math.log %arg0 : vector<3xf32>
// CHECK: %[[ONE:.+]] = spirv.Constant dense<1.000000e+00> : vector<3xf32>
// CHECK: %[[ADDONE:.+]] = spirv.FAdd %[[ONE]], %{{.+}}
// CHECK: spirv.CL.log %[[ADDONE]]
%5 = math.log1p %arg0 : vector<3xf32>
// CHECK: spirv.CL.log2 %{{.*}}: vector<3xf32>
%6 = math.log2 %arg0 : vector<3xf32>
// CHECK: spirv.CL.log10 %{{.*}}: vector<3xf32>
%7 = math.log10 %arg0 : vector<3xf32>
// CHECK: spirv.CL.exp2 %{{.*}}: vector<3xf32>
%exp2_vec = math.exp2 %arg0 : vector<3xf32>
// CHECK: spirv.CL.rint %{{.*}}: vector<3xf32>
%8 = math.roundeven %arg0 : vector<3xf32>
// CHECK: spirv.CL.rsqrt %{{.*}}: vector<3xf32>
%9 = math.rsqrt %arg0 : vector<3xf32>
// CHECK: spirv.CL.sqrt %{{.*}}: vector<3xf32>
%10 = math.sqrt %arg0 : vector<3xf32>
// CHECK: spirv.CL.tanh %{{.*}}: vector<3xf32>
%11 = math.tanh %arg0 : vector<3xf32>
// CHECK: spirv.CL.sin %{{.*}}: vector<3xf32>
%12 = math.sin %arg0 : vector<3xf32>
// CHECK: spirv.CL.tan %{{.*}}: vector<3xf32>
%13 = math.tan %arg0 : vector<3xf32>
// CHECK: spirv.CL.asin %{{.*}}: vector<3xf32>
%14 = math.asin %arg0 : vector<3xf32>
// CHECK: spirv.CL.acos %{{.*}}: vector<3xf32>
%15 = math.acos %arg0 : vector<3xf32>
// CHECK: spirv.CL.sinh %{{.*}}: vector<3xf32>
%16 = math.sinh %arg0 : vector<3xf32>
// CHECK: spirv.CL.cosh %{{.*}}: vector<3xf32>
%17 = math.cosh %arg0 : vector<3xf32>
// CHECK: spirv.CL.asinh %{{.*}}: vector<3xf32>
%18 = math.asinh %arg0 : vector<3xf32>
// CHECK: spirv.CL.acosh %{{.*}}: vector<3xf32>
%19 = math.acosh %arg0 : vector<3xf32>
// CHECK: spirv.CL.atanh %{{.*}}: vector<3xf32>
%20 = math.atanh %arg0 : vector<3xf32>
// CHECK: spirv.CL.trunc %{{.*}}: vector<3xf32>
%21 = math.trunc %arg0 : vector<3xf32>
return
}
// CHECK-LABEL: @float32_binary_scalar
func.func @float32_binary_scalar(%lhs: f32, %rhs: f32) {
// CHECK: spirv.CL.atan2 %{{.*}}: f32
%0 = math.atan2 %lhs, %rhs : f32
// CHECK: spirv.CL.pow %{{.*}}: f32
%1 = math.powf %lhs, %rhs : f32
return
}
// CHECK-LABEL: @float32_binary_vector
func.func @float32_binary_vector(%lhs: vector<4xf32>, %rhs: vector<4xf32>) {
// CHECK: spirv.CL.atan2 %{{.*}}: vector<4xf32>
%0 = math.atan2 %lhs, %rhs : vector<4xf32>
// CHECK: spirv.CL.pow %{{.*}}: vector<4xf32>
%1 = math.powf %lhs, %rhs : vector<4xf32>
return
}
// CHECK-LABEL: @fpowi_scalar
func.func @fpowi_scalar(%base: f32, %power: i32) -> f32 {
// CHECK: spirv.CL.pown %{{.*}}, %{{.*}} : f32, i32 -> f32
%0 = math.fpowi %base, %power : f32, i32
return %0 : f32
}
// CHECK-LABEL: @fpowi_vector
func.func @fpowi_vector(%base: vector<4xf32>, %power: vector<4xi32>) -> vector<4xf32> {
// CHECK: spirv.CL.pown %{{.*}}, %{{.*}} : vector<4xf32>, vector<4xi32> -> vector<4xf32>
%0 = math.fpowi %base, %power : vector<4xf32>, vector<4xi32>
return %0 : vector<4xf32>
}
// CHECK-LABEL: @float32_ternary_scalar
func.func @float32_ternary_scalar(%a: f32, %b: f32, %c: f32) {
// CHECK: spirv.CL.fma %{{.*}}: f32
%0 = math.fma %a, %b, %c : f32
return
}
// CHECK-LABEL: @float32_ternary_vector
func.func @float32_ternary_vector(%a: vector<4xf32>, %b: vector<4xf32>,
%c: vector<4xf32>) {
// CHECK: spirv.CL.fma %{{.*}}: vector<4xf32>
%0 = math.fma %a, %b, %c : vector<4xf32>
return
}
// CHECK-LABEL: @int_unary
func.func @int_unary(%arg0: i32) {
// CHECK: spirv.CL.s_abs %{{.*}}
%0 = math.absi %arg0 : i32
// CHECK: spirv.CL.clz %{{.*}} : i32
%1 = math.ctlz %arg0 : i32
return
}
// CHECK-LABEL: @int_unary_vector
func.func @int_unary_vector(%arg0: vector<2xi32>) {
// CHECK: spirv.CL.clz %{{.*}} : vector<2xi32>
%0 = math.ctlz %arg0 : vector<2xi32>
return
}
} // end module
// -----
module attributes {
spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [Shader], []>, #spirv.resource_limits<>>
} {
// 2-D vectors are not supported.
// CHECK-LABEL: @vector_2d
func.func @vector_2d(%arg0: vector<2x2xf32>) {
// CHECK-NEXT: math.atan {{.+}} : vector<2x2xf32>
%0 = math.atan %arg0 : vector<2x2xf32>
// CHECK-NEXT: math.cos {{.+}} : vector<2x2xf32>
%1 = math.cos %arg0 : vector<2x2xf32>
// CHECK-NEXT: math.exp {{.+}} : vector<2x2xf32>
%2 = math.exp %arg0 : vector<2x2xf32>
// CHECK-NEXT: math.absf {{.+}} : vector<2x2xf32>
%3 = math.absf %arg0 : vector<2x2xf32>
// CHECK-NEXT: math.ceil {{.+}} : vector<2x2xf32>
%4 = math.ceil %arg0 : vector<2x2xf32>
// CHECK-NEXT: math.floor {{.+}} : vector<2x2xf32>
%5 = math.floor %arg0 : vector<2x2xf32>
// CHECK-NEXT: math.powf {{.+}}, {{%.+}} : vector<2x2xf32>
%6 = math.powf %arg0, %arg0 : vector<2x2xf32>
// CHECK-NEXT: return
return
}
// Tensors are not supported.
// CHECK-LABEL: @tensor_1d
func.func @tensor_1d(%arg0: tensor<2xf32>) {
// CHECK-NEXT: math.atan {{.+}} : tensor<2xf32>
%0 = math.atan %arg0 : tensor<2xf32>
// CHECK-NEXT: math.cos {{.+}} : tensor<2xf32>
%1 = math.cos %arg0 : tensor<2xf32>
// CHECK-NEXT: math.exp {{.+}} : tensor<2xf32>
%2 = math.exp %arg0 : tensor<2xf32>
// CHECK-NEXT: math.absf {{.+}} : tensor<2xf32>
%3 = math.absf %arg0 : tensor<2xf32>
// CHECK-NEXT: math.ceil {{.+}} : tensor<2xf32>
%4 = math.ceil %arg0 : tensor<2xf32>
// CHECK-NEXT: math.floor {{.+}} : tensor<2xf32>
%5 = math.floor %arg0 : tensor<2xf32>
// CHECK-NEXT: math.powf {{.+}}, {{%.+}} : tensor<2xf32>
%6 = math.powf %arg0, %arg0 : tensor<2xf32>
// CHECK-NEXT: return
return
}
} // end module
Reference in new issue View Git Blame Copy Permalink