32389d0c2e
Also, it seems Khronos has changed html spec format so small adjustment to script was needed. Base op parsing is also probably broken. Differential Revision: https://reviews.llvm.org/D119678
97 lines
3.2 KiB
MLIR
97 lines
3.2 KiB
MLIR
// RUN: mlir-opt -split-input-file -convert-math-to-spirv -verify-diagnostics %s -o - | FileCheck %s
|
|
|
|
module attributes { spv.target_env = #spv.target_env<#spv.vce<v1.0, [Kernel], []>, {}> } {
|
|
|
|
// CHECK-LABEL: @float32_unary_scalar
|
|
func @float32_unary_scalar(%arg0: f32) {
|
|
// CHECK: spv.OCL.cos %{{.*}}: f32
|
|
%0 = math.cos %arg0 : f32
|
|
// CHECK: spv.OCL.exp %{{.*}}: f32
|
|
%1 = math.exp %arg0 : f32
|
|
// CHECK: %[[EXP:.+]] = spv.OCL.exp %arg0
|
|
// CHECK: %[[ONE:.+]] = spv.Constant 1.000000e+00 : f32
|
|
// CHECK: spv.FSub %[[EXP]], %[[ONE]]
|
|
%2 = math.expm1 %arg0 : f32
|
|
// CHECK: spv.OCL.log %{{.*}}: f32
|
|
%3 = math.log %arg0 : f32
|
|
// CHECK: %[[ONE:.+]] = spv.Constant 1.000000e+00 : f32
|
|
// CHECK: %[[ADDONE:.+]] = spv.FAdd %[[ONE]], %{{.+}}
|
|
// CHECK: spv.OCL.log %[[ADDONE]]
|
|
%4 = math.log1p %arg0 : f32
|
|
// CHECK: spv.OCL.rsqrt %{{.*}}: f32
|
|
%5 = math.rsqrt %arg0 : f32
|
|
// CHECK: spv.OCL.sqrt %{{.*}}: f32
|
|
%6 = math.sqrt %arg0 : f32
|
|
// CHECK: spv.OCL.tanh %{{.*}}: f32
|
|
%7 = math.tanh %arg0 : f32
|
|
// CHECK: spv.OCL.sin %{{.*}}: f32
|
|
%8 = math.sin %arg0 : f32
|
|
// CHECK: spv.OCL.fabs %{{.*}}: f32
|
|
%9 = math.abs %arg0 : f32
|
|
// CHECK: spv.OCL.ceil %{{.*}}: f32
|
|
%10 = math.ceil %arg0 : f32
|
|
// CHECK: spv.OCL.floor %{{.*}}: f32
|
|
%11 = math.floor %arg0 : f32
|
|
// CHECK: spv.OCL.erf %{{.*}}: f32
|
|
%12 = math.erf %arg0 : f32
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @float32_unary_vector
|
|
func @float32_unary_vector(%arg0: vector<3xf32>) {
|
|
// CHECK: spv.OCL.cos %{{.*}}: vector<3xf32>
|
|
%0 = math.cos %arg0 : vector<3xf32>
|
|
// CHECK: spv.OCL.exp %{{.*}}: vector<3xf32>
|
|
%1 = math.exp %arg0 : vector<3xf32>
|
|
// CHECK: %[[EXP:.+]] = spv.OCL.exp %arg0
|
|
// CHECK: %[[ONE:.+]] = spv.Constant dense<1.000000e+00> : vector<3xf32>
|
|
// CHECK: spv.FSub %[[EXP]], %[[ONE]]
|
|
%2 = math.expm1 %arg0 : vector<3xf32>
|
|
// CHECK: spv.OCL.log %{{.*}}: vector<3xf32>
|
|
%3 = math.log %arg0 : vector<3xf32>
|
|
// CHECK: %[[ONE:.+]] = spv.Constant dense<1.000000e+00> : vector<3xf32>
|
|
// CHECK: %[[ADDONE:.+]] = spv.FAdd %[[ONE]], %{{.+}}
|
|
// CHECK: spv.OCL.log %[[ADDONE]]
|
|
%4 = math.log1p %arg0 : vector<3xf32>
|
|
// CHECK: spv.OCL.rsqrt %{{.*}}: vector<3xf32>
|
|
%5 = math.rsqrt %arg0 : vector<3xf32>
|
|
// CHECK: spv.OCL.sqrt %{{.*}}: vector<3xf32>
|
|
%6 = math.sqrt %arg0 : vector<3xf32>
|
|
// CHECK: spv.OCL.tanh %{{.*}}: vector<3xf32>
|
|
%7 = math.tanh %arg0 : vector<3xf32>
|
|
// CHECK: spv.OCL.sin %{{.*}}: vector<3xf32>
|
|
%8 = math.sin %arg0 : vector<3xf32>
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @float32_binary_scalar
|
|
func @float32_binary_scalar(%lhs: f32, %rhs: f32) {
|
|
// CHECK: spv.OCL.pow %{{.*}}: f32
|
|
%0 = math.powf %lhs, %rhs : f32
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @float32_binary_vector
|
|
func @float32_binary_vector(%lhs: vector<4xf32>, %rhs: vector<4xf32>) {
|
|
// CHECK: spv.OCL.pow %{{.*}}: vector<4xf32>
|
|
%0 = math.powf %lhs, %rhs : vector<4xf32>
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @float32_ternary_scalar
|
|
func @float32_ternary_scalar(%a: f32, %b: f32, %c: f32) {
|
|
// CHECK: spv.OCL.fma %{{.*}}: f32
|
|
%0 = math.fma %a, %b, %c : f32
|
|
return
|
|
}
|
|
|
|
// CHECK-LABEL: @float32_ternary_vector
|
|
func @float32_ternary_vector(%a: vector<4xf32>, %b: vector<4xf32>,
|
|
%c: vector<4xf32>) {
|
|
// CHECK: spv.OCL.fma %{{.*}}: vector<4xf32>
|
|
%0 = math.fma %a, %b, %c : vector<4xf32>
|
|
return
|
|
}
|
|
|
|
} // end module
|