335 lines · plain
1// RUN: mlir-opt -split-input-file -convert-math-to-spirv -verify-diagnostics %s -o - | FileCheck %s2 3module attributes {4 spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [Shader], []>, #spirv.resource_limits<>>5} {6 7// CHECK-LABEL: @float32_unary_scalar8func.func @float32_unary_scalar(%arg0: f32) {9 // CHECK: spirv.GL.Atan %{{.*}}: f3210 %0 = math.atan %arg0 : f3211 // CHECK: spirv.GL.Cos %{{.*}}: f3212 %1 = math.cos %arg0 : f3213 // CHECK: spirv.GL.Exp %{{.*}}: f3214 %2 = math.exp %arg0 : f3215 // CHECK: %[[EXP:.+]] = spirv.GL.Exp %arg016 // CHECK: %[[ONE:.+]] = spirv.Constant 1.000000e+00 : f3217 // CHECK: spirv.FSub %[[EXP]], %[[ONE]]18 %3 = math.expm1 %arg0 : f3219 // CHECK: spirv.GL.Log %{{.*}}: f3220 %4 = math.log %arg0 : f3221 // CHECK: %[[ONE:.+]] = spirv.Constant 1.000000e+00 : f3222 // CHECK: %[[ADDONE:.+]] = spirv.FAdd %[[ONE]], %{{.+}}23 // CHECK: spirv.GL.Log %[[ADDONE]]24 %5 = math.log1p %arg0 : f3225 // CHECK: %[[LOG2_RECIPROCAL:.+]] = spirv.Constant 1.44269502 : f3226 // CHECK: %[[LOG0:.+]] = spirv.GL.Log {{.+}}27 // CHECK: spirv.FMul %[[LOG0]], %[[LOG2_RECIPROCAL]]28 %6 = math.log2 %arg0 : f3229 // CHECK: %[[LOG10_RECIPROCAL:.+]] = spirv.Constant 0.434294492 : f3230 // CHECK: %[[LOG1:.+]] = spirv.GL.Log {{.+}}31 // CHECK: spirv.FMul %[[LOG1]], %[[LOG10_RECIPROCAL]]32 %7 = math.log10 %arg0 : f3233 // CHECK: spirv.GL.RoundEven %{{.*}}: f3234 %8 = math.roundeven %arg0 : f3235 // CHECK: spirv.GL.InverseSqrt %{{.*}}: f3236 %9 = math.rsqrt %arg0 : f3237 // CHECK: spirv.GL.Sqrt %{{.*}}: f3238 %10 = math.sqrt %arg0 : f3239 // CHECK: spirv.GL.Tanh %{{.*}}: f3240 %11 = math.tanh %arg0 : f3241 // CHECK: spirv.GL.Sin %{{.*}}: f3242 %12 = math.sin %arg0 : f3243 // CHECK: spirv.GL.FAbs %{{.*}}: f3244 %13 = math.absf %arg0 : f3245 // CHECK: spirv.GL.Ceil %{{.*}}: f3246 %14 = math.ceil %arg0 : f3247 // CHECK: spirv.GL.Floor %{{.*}}: f3248 %15 = math.floor %arg0 : f3249 // CHECK: spirv.GL.Tan %{{.*}}: f3250 %16 = math.tan %arg0 : f3251 // CHECK: spirv.GL.Asin %{{.*}}: f3252 %17 = math.asin %arg0 : f3253 // CHECK: spirv.GL.Acos %{{.*}}: f3254 %18 = math.acos %arg0 : f3255 // CHECK: spirv.GL.Sinh %{{.*}}: f3256 %19 = math.sinh %arg0 : f3257 // CHECK: spirv.GL.Cosh %{{.*}}: f3258 %20 = math.cosh %arg0 : f3259 // CHECK: spirv.GL.Asinh %{{.*}}: f3260 %21 = math.asinh %arg0 : f3261 // CHECK: spirv.GL.Acosh %{{.*}}: f3262 %22 = math.acosh %arg0 : f3263 // CHECK: spirv.GL.Atanh %{{.*}}: f3264 %23 = math.atanh %arg0 : f3265 return66}67 68// CHECK-LABEL: @float32_unary_vector69func.func @float32_unary_vector(%arg0: vector<3xf32>) {70 // CHECK: spirv.GL.Atan %{{.*}}: vector<3xf32>71 %0 = math.atan %arg0 : vector<3xf32>72 // CHECK: spirv.GL.Cos %{{.*}}: vector<3xf32>73 %1 = math.cos %arg0 : vector<3xf32>74 // CHECK: spirv.GL.Exp %{{.*}}: vector<3xf32>75 %2 = math.exp %arg0 : vector<3xf32>76 // CHECK: %[[EXP:.+]] = spirv.GL.Exp %arg077 // CHECK: %[[ONE:.+]] = spirv.Constant dense<1.000000e+00> : vector<3xf32>78 // CHECK: spirv.FSub %[[EXP]], %[[ONE]]79 %3 = math.expm1 %arg0 : vector<3xf32>80 // CHECK: spirv.GL.Log %{{.*}}: vector<3xf32>81 %4 = math.log %arg0 : vector<3xf32>82 // CHECK: %[[ONE:.+]] = spirv.Constant dense<1.000000e+00> : vector<3xf32>83 // CHECK: %[[ADDONE:.+]] = spirv.FAdd %[[ONE]], %{{.+}}84 // CHECK: spirv.GL.Log %[[ADDONE]]85 %5 = math.log1p %arg0 : vector<3xf32>86 // CHECK: %[[LOG2_RECIPROCAL:.+]] = spirv.Constant dense<1.44269502> : vector<3xf32>87 // CHECK: %[[LOG0:.+]] = spirv.GL.Log {{.+}}88 // CHECK: spirv.FMul %[[LOG0]], %[[LOG2_RECIPROCAL]]89 %6 = math.log2 %arg0 : vector<3xf32>90 // CHECK: %[[LOG10_RECIPROCAL:.+]] = spirv.Constant dense<0.434294492> : vector<3xf32>91 // CHECK: %[[LOG1:.+]] = spirv.GL.Log {{.+}}92 // CHECK: spirv.FMul %[[LOG1]], %[[LOG10_RECIPROCAL]]93 %7 = math.log10 %arg0 : vector<3xf32>94 // CHECK: spirv.GL.RoundEven %{{.*}}: vector<3xf32>95 %8 = math.roundeven %arg0 : vector<3xf32>96 // CHECK: spirv.GL.InverseSqrt %{{.*}}: vector<3xf32>97 %9 = math.rsqrt %arg0 : vector<3xf32>98 // CHECK: spirv.GL.Sqrt %{{.*}}: vector<3xf32>99 %10 = math.sqrt %arg0 : vector<3xf32>100 // CHECK: spirv.GL.Tanh %{{.*}}: vector<3xf32>101 %11 = math.tanh %arg0 : vector<3xf32>102 // CHECK: spirv.GL.Sin %{{.*}}: vector<3xf32>103 %12 = math.sin %arg0 : vector<3xf32>104 // CHECK: spirv.GL.Tan %{{.*}}: vector<3xf32>105 %13 = math.tan %arg0 : vector<3xf32>106 // CHECK: spirv.GL.Asin %{{.*}}: vector<3xf32>107 %14 = math.asin %arg0 : vector<3xf32>108 // CHECK: spirv.GL.Acos %{{.*}}: vector<3xf32>109 %15 = math.acos %arg0 : vector<3xf32>110 // CHECK: spirv.GL.Sinh %{{.*}}: vector<3xf32>111 %16 = math.sinh %arg0 : vector<3xf32>112 // CHECK: spirv.GL.Cosh %{{.*}}: vector<3xf32>113 %17 = math.cosh %arg0 : vector<3xf32>114 // CHECK: spirv.GL.Asinh %{{.*}}: vector<3xf32>115 %18 = math.asinh %arg0 : vector<3xf32>116 // CHECK: spirv.GL.Acosh %{{.*}}: vector<3xf32>117 %19 = math.acosh %arg0 : vector<3xf32>118 // CHECK: spirv.GL.Atanh %{{.*}}: vector<3xf32>119 %20 = math.atanh %arg0 : vector<3xf32>120 return121}122 123// CHECK-LABEL: @float32_ternary_scalar124func.func @float32_ternary_scalar(%a: f32, %b: f32, %c: f32) {125 // CHECK: spirv.GL.Fma %{{.*}}: f32126 %0 = math.fma %a, %b, %c : f32127 return128}129 130// CHECK-LABEL: @float32_ternary_vector131func.func @float32_ternary_vector(%a: vector<4xf32>, %b: vector<4xf32>,132 %c: vector<4xf32>) {133 // CHECK: spirv.GL.Fma %{{.*}}: vector<4xf32>134 %0 = math.fma %a, %b, %c : vector<4xf32>135 return136}137 138// CHECK-LABEL: @int_unary139func.func @int_unary(%arg0: i32) {140 // CHECK: spirv.GL.SAbs %{{.*}}141 %0 = math.absi %arg0 : i32142 return143}144 145// CHECK-LABEL: @ctlz_scalar146// CHECK-SAME: (%[[VAL:.+]]: i32)147func.func @ctlz_scalar(%val: i32) -> i32 {148 // CHECK-DAG: %[[V1:.+]] = spirv.Constant 1 : i32149 // CHECK-DAG: %[[V31:.+]] = spirv.Constant 31 : i32150 // CHECK-DAG: %[[V32:.+]] = spirv.Constant 32 : i32151 // CHECK: %[[MSB:.+]] = spirv.GL.FindUMsb %[[VAL]] : i32152 // CHECK: %[[SUB1:.+]] = spirv.ISub %[[V31]], %[[MSB]] : i32153 // CHECK: %[[SUB2:.+]] = spirv.ISub %[[V32]], %[[VAL]] : i32154 // CHECK: %[[CMP:.+]] = spirv.ULessThanEqual %[[VAL]], %[[V1]] : i32155 // CHECK: %[[R:.+]] = spirv.Select %[[CMP]], %[[SUB2]], %[[SUB1]] : i1, i32156 // CHECK: return %[[R]]157 %0 = math.ctlz %val : i32158 return %0 : i32159}160 161// CHECK-LABEL: @ctlz_vector1162func.func @ctlz_vector1(%val: vector<1xi32>) -> vector<1xi32> {163 // CHECK: spirv.GL.FindUMsb164 // CHECK: spirv.ISub165 // CHECK: spirv.ULessThanEqual166 // CHECK: spirv.Select167 %0 = math.ctlz %val : vector<1xi32>168 return %0 : vector<1xi32>169}170 171// CHECK-LABEL: @ctlz_vector2172// CHECK-SAME: (%[[VAL:.+]]: vector<2xi32>)173func.func @ctlz_vector2(%val: vector<2xi32>) -> vector<2xi32> {174 // CHECK-DAG: %[[V1:.+]] = spirv.Constant dense<1> : vector<2xi32>175 // CHECK-DAG: %[[V31:.+]] = spirv.Constant dense<31> : vector<2xi32>176 // CHECK-DAG: %[[V32:.+]] = spirv.Constant dense<32> : vector<2xi32>177 // CHECK: %[[MSB:.+]] = spirv.GL.FindUMsb %[[VAL]] : vector<2xi32>178 // CHECK: %[[SUB1:.+]] = spirv.ISub %[[V31]], %[[MSB]] : vector<2xi32>179 // CHECK: %[[SUB2:.+]] = spirv.ISub %[[V32]], %[[VAL]] : vector<2xi32>180 // CHECK: %[[CMP:.+]] = spirv.ULessThanEqual %[[VAL]], %[[V1]] : vector<2xi32>181 // CHECK: %[[R:.+]] = spirv.Select %[[CMP]], %[[SUB2]], %[[SUB1]] : vector<2xi1>, vector<2xi32>182 %0 = math.ctlz %val : vector<2xi32>183 return %0 : vector<2xi32>184}185 186// CHECK-LABEL: @powf_scalar187// CHECK-SAME: (%[[LHS:.+]]: f32, %[[RHS:.+]]: f32)188func.func @powf_scalar(%lhs: f32, %rhs: f32) -> f32 {189 // CHECK: %[[F0:.+]] = spirv.Constant 0.000000e+00 : f32190 // CHECK: %[[LT:.+]] = spirv.FOrdLessThan %[[LHS]], %[[F0]] : f32191 // CHECK: %[[F1:.+]] = spirv.Constant 1.000000e+00 : f32192 // CHECK: %[[REM:.+]] = spirv.FRem %[[RHS]], %[[F1]] : f32193 // CHECK: %[[IS_FRACTION:.+]] = spirv.FOrdNotEqual %[[REM]], %[[F0]] : f32194 // CHECK: %[[AND:.+]] = spirv.LogicalAnd %[[IS_FRACTION]], %[[LT]] : i1195 // CHECK: %[[NAN:.+]] = spirv.Constant 0x7FC00000 : f32196 // CHECK: %[[NEW_LHS:.+]] = spirv.Select %[[AND]], %[[NAN]], %[[LHS]] : i1, f32197 // CHECK: %[[ABS:.+]] = spirv.GL.FAbs %[[NEW_LHS]] : f32198 // CHECK: %[[IRHS:.+]] = spirv.ConvertFToS199 // CHECK: %[[CST1:.+]] = spirv.Constant 1 : i32200 // CHECK: %[[REM:.+]] = spirv.BitwiseAnd %[[IRHS]]201 // CHECK: %[[ODD:.+]] = spirv.IEqual %[[REM]], %[[CST1]] : i32202 // CHECK: %[[POW:.+]] = spirv.GL.Pow %[[ABS]], %[[RHS]] : f32203 // CHECK: %[[NEG:.+]] = spirv.FNegate %[[POW]] : f32204 // CHECK: %[[SNEG:.+]] = spirv.LogicalAnd %[[LT]], %[[ODD]] : i1205 // CHECK: %[[SEL:.+]] = spirv.Select %[[SNEG]], %[[NEG]], %[[POW]] : i1, f32206 %0 = math.powf %lhs, %rhs : f32207 // CHECK: return %[[SEL]]208 return %0: f32209}210 211// CHECK-LABEL: @powf_vector212func.func @powf_vector(%lhs: vector<4xf32>, %rhs: vector<4xf32>) -> vector<4xf32> {213 // CHECK: spirv.FOrdLessThan214 // CHECK: spirv.FRem215 // CHECK: spirv.FOrdNotEqual216 // CHECK: spirv.LogicalAnd217 // CHECK: spirv.Select218 // CHECK: spirv.GL.FAbs219 // CHECK: spirv.BitwiseAnd %{{.*}} : vector<4xi32>220 // CHECK: spirv.IEqual %{{.*}} : vector<4xi32>221 // CHECK: spirv.GL.Pow %{{.*}}: vector<4xf32>222 // CHECK: spirv.FNegate223 // CHECK: spirv.Select224 %0 = math.powf %lhs, %rhs : vector<4xf32>225 return %0: vector<4xf32>226}227 228// CHECK-LABEL: @round_scalar229func.func @round_scalar(%x: f32) -> f32 {230 // CHECK: %[[ZERO:.+]] = spirv.Constant 0.000000e+00231 // CHECK: %[[ONE:.+]] = spirv.Constant 1.000000e+00232 // CHECK: %[[HALF:.+]] = spirv.Constant 5.000000e-01233 // CHECK: %[[ABS:.+]] = spirv.GL.FAbs %arg0234 // CHECK: %[[FLOOR:.+]] = spirv.GL.Floor %[[ABS]]235 // CHECK: %[[SUB:.+]] = spirv.FSub %[[ABS]], %[[FLOOR]]236 // CHECK: %[[GE:.+]] = spirv.FOrdGreaterThanEqual %[[SUB]], %[[HALF]]237 // CHECK: %[[SEL:.+]] = spirv.Select %[[GE]], %[[ONE]], %[[ZERO]]238 // CHECK: %[[ADD:.+]] = spirv.FAdd %[[FLOOR]], %[[SEL]]239 // CHECK: %[[BITCAST:.+]] = spirv.Bitcast %[[ADD]]240 %0 = math.round %x : f32241 return %0: f32242}243 244// CHECK-LABEL: @round_vector245func.func @round_vector(%x: vector<4xf32>) -> vector<4xf32> {246 // CHECK: %[[ZERO:.+]] = spirv.Constant dense<0.000000e+00>247 // CHECK: %[[ONE:.+]] = spirv.Constant dense<1.000000e+00>248 // CHECK: %[[HALF:.+]] = spirv.Constant dense<5.000000e-01>249 // CHECK: %[[ABS:.+]] = spirv.GL.FAbs %arg0250 // CHECK: %[[FLOOR:.+]] = spirv.GL.Floor %[[ABS]]251 // CHECK: %[[SUB:.+]] = spirv.FSub %[[ABS]], %[[FLOOR]]252 // CHECK: %[[GE:.+]] = spirv.FOrdGreaterThanEqual %[[SUB]], %[[HALF]]253 // CHECK: %[[SEL:.+]] = spirv.Select %[[GE]], %[[ONE]], %[[ZERO]]254 // CHECK: %[[ADD:.+]] = spirv.FAdd %[[FLOOR]], %[[SEL]]255 // CHECK: %[[BITCAST:.+]] = spirv.Bitcast %[[ADD]]256 %0 = math.round %x : vector<4xf32>257 return %0: vector<4xf32>258}259 260} // end module261 262// -----263 264module attributes {265 spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [Shader, Int64, Int16], []>, #spirv.resource_limits<>>266} {267 268// CHECK-LABEL: @ctlz_scalar269func.func @ctlz_scalar(%val: i64) -> i64 {270 // CHECK: math.ctlz271 %0 = math.ctlz %val : i64272 return %0 : i64273}274 275// CHECK-LABEL: @ctlz_vector2276func.func @ctlz_vector2(%val: vector<2xi16>) -> vector<2xi16> {277 // CHECK: math.ctlz278 %0 = math.ctlz %val : vector<2xi16>279 return %0 : vector<2xi16>280}281 282} // end module283 284// -----285 286module attributes {287 spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [Shader], []>, #spirv.resource_limits<>>288} {289 290// 2-D vectors are not supported.291 292// CHECK-LABEL: @vector_2d293func.func @vector_2d(%arg0: vector<2x2xf32>) {294 // CHECK-NEXT: math.atan {{.+}} : vector<2x2xf32>295 %0 = math.atan %arg0 : vector<2x2xf32>296 // CHECK-NEXT: math.cos {{.+}} : vector<2x2xf32>297 %1 = math.cos %arg0 : vector<2x2xf32>298 // CHECK-NEXT: math.exp {{.+}} : vector<2x2xf32>299 %2 = math.exp %arg0 : vector<2x2xf32>300 // CHECK-NEXT: math.absf {{.+}} : vector<2x2xf32>301 %3 = math.absf %arg0 : vector<2x2xf32>302 // CHECK-NEXT: math.ceil {{.+}} : vector<2x2xf32>303 %4 = math.ceil %arg0 : vector<2x2xf32>304 // CHECK-NEXT: math.floor {{.+}} : vector<2x2xf32>305 %5 = math.floor %arg0 : vector<2x2xf32>306 // CHECK-NEXT: math.powf {{.+}}, {{%.+}} : vector<2x2xf32>307 %6 = math.powf %arg0, %arg0 : vector<2x2xf32>308 // CHECK-NEXT: return309 return310}311 312// Tensors are not supported.313 314// CHECK-LABEL: @tensor_1d315func.func @tensor_1d(%arg0: tensor<2xf32>) {316 // CHECK-NEXT: math.atan {{.+}} : tensor<2xf32>317 %0 = math.atan %arg0 : tensor<2xf32>318 // CHECK-NEXT: math.cos {{.+}} : tensor<2xf32>319 %1 = math.cos %arg0 : tensor<2xf32>320 // CHECK-NEXT: math.exp {{.+}} : tensor<2xf32>321 %2 = math.exp %arg0 : tensor<2xf32>322 // CHECK-NEXT: math.absf {{.+}} : tensor<2xf32>323 %3 = math.absf %arg0 : tensor<2xf32>324 // CHECK-NEXT: math.ceil {{.+}} : tensor<2xf32>325 %4 = math.ceil %arg0 : tensor<2xf32>326 // CHECK-NEXT: math.floor {{.+}} : tensor<2xf32>327 %5 = math.floor %arg0 : tensor<2xf32>328 // CHECK-NEXT: math.powf {{.+}}, {{%.+}} : tensor<2xf32>329 %6 = math.powf %arg0, %arg0 : tensor<2xf32>330 // CHECK-NEXT: return331 return332}333 334} // end module335