193 lines · plain
1// RUN: mlir-opt -convert-spirv-to-llvm %s | FileCheck %s2 3//===----------------------------------------------------------------------===//4// spirv.GL.Ceil5//===----------------------------------------------------------------------===//6 7// CHECK-LABEL: @ceil8spirv.func @ceil(%arg0: f32, %arg1: vector<3xf16>) "None" {9 // CHECK: llvm.intr.ceil(%{{.*}}) : (f32) -> f3210 %0 = spirv.GL.Ceil %arg0 : f3211 // CHECK: llvm.intr.ceil(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>12 %1 = spirv.GL.Ceil %arg1 : vector<3xf16>13 spirv.Return14}15 16//===----------------------------------------------------------------------===//17// spirv.GL.Cos18//===----------------------------------------------------------------------===//19 20// CHECK-LABEL: @cos21spirv.func @cos(%arg0: f32, %arg1: vector<3xf16>) "None" {22 // CHECK: llvm.intr.cos(%{{.*}}) : (f32) -> f3223 %0 = spirv.GL.Cos %arg0 : f3224 // CHECK: llvm.intr.cos(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>25 %1 = spirv.GL.Cos %arg1 : vector<3xf16>26 spirv.Return27}28 29//===----------------------------------------------------------------------===//30// spirv.GL.Exp31//===----------------------------------------------------------------------===//32 33// CHECK-LABEL: @exp34spirv.func @exp(%arg0: f32, %arg1: vector<3xf16>) "None" {35 // CHECK: llvm.intr.exp(%{{.*}}) : (f32) -> f3236 %0 = spirv.GL.Exp %arg0 : f3237 // CHECK: llvm.intr.exp(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>38 %1 = spirv.GL.Exp %arg1 : vector<3xf16>39 spirv.Return40}41 42//===----------------------------------------------------------------------===//43// spirv.GL.FAbs44//===----------------------------------------------------------------------===//45 46// CHECK-LABEL: @fabs47spirv.func @fabs(%arg0: f32, %arg1: vector<3xf16>) "None" {48 // CHECK: llvm.intr.fabs(%{{.*}}) : (f32) -> f3249 %0 = spirv.GL.FAbs %arg0 : f3250 // CHECK: llvm.intr.fabs(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>51 %1 = spirv.GL.FAbs %arg1 : vector<3xf16>52 spirv.Return53}54 55//===----------------------------------------------------------------------===//56// spirv.GL.Floor57//===----------------------------------------------------------------------===//58 59// CHECK-LABEL: @floor60spirv.func @floor(%arg0: f32, %arg1: vector<3xf16>) "None" {61 // CHECK: llvm.intr.floor(%{{.*}}) : (f32) -> f3262 %0 = spirv.GL.Floor %arg0 : f3263 // CHECK: llvm.intr.floor(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>64 %1 = spirv.GL.Floor %arg1 : vector<3xf16>65 spirv.Return66}67 68//===----------------------------------------------------------------------===//69// spirv.GL.FMax70//===----------------------------------------------------------------------===//71 72// CHECK-LABEL: @fmax73spirv.func @fmax(%arg0: f32, %arg1: vector<3xf16>) "None" {74 // CHECK: llvm.intr.maxnum(%{{.*}}, %{{.*}}) : (f32, f32) -> f3275 %0 = spirv.GL.FMax %arg0, %arg0 : f3276 // CHECK: llvm.intr.maxnum(%{{.*}}, %{{.*}}) : (vector<3xf16>, vector<3xf16>) -> vector<3xf16>77 %1 = spirv.GL.FMax %arg1, %arg1 : vector<3xf16>78 spirv.Return79}80 81//===----------------------------------------------------------------------===//82// spirv.GL.FMin83//===----------------------------------------------------------------------===//84 85// CHECK-LABEL: @fmin86spirv.func @fmin(%arg0: f32, %arg1: vector<3xf16>) "None" {87 // CHECK: llvm.intr.minnum(%{{.*}}, %{{.*}}) : (f32, f32) -> f3288 %0 = spirv.GL.FMin %arg0, %arg0 : f3289 // CHECK: llvm.intr.minnum(%{{.*}}, %{{.*}}) : (vector<3xf16>, vector<3xf16>) -> vector<3xf16>90 %1 = spirv.GL.FMin %arg1, %arg1 : vector<3xf16>91 spirv.Return92}93 94//===----------------------------------------------------------------------===//95// spirv.GL.Log96//===----------------------------------------------------------------------===//97 98// CHECK-LABEL: @log99spirv.func @log(%arg0: f32, %arg1: vector<3xf16>) "None" {100 // CHECK: llvm.intr.log(%{{.*}}) : (f32) -> f32101 %0 = spirv.GL.Log %arg0 : f32102 // CHECK: llvm.intr.log(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>103 %1 = spirv.GL.Log %arg1 : vector<3xf16>104 spirv.Return105}106 107//===----------------------------------------------------------------------===//108// spirv.GL.Sin109//===----------------------------------------------------------------------===//110 111// CHECK-LABEL: @sin112spirv.func @sin(%arg0: f32, %arg1: vector<3xf16>) "None" {113 // CHECK: llvm.intr.sin(%{{.*}}) : (f32) -> f32114 %0 = spirv.GL.Sin %arg0 : f32115 // CHECK: llvm.intr.sin(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>116 %1 = spirv.GL.Sin %arg1 : vector<3xf16>117 spirv.Return118}119 120//===----------------------------------------------------------------------===//121// spirv.GL.SMax122//===----------------------------------------------------------------------===//123 124// CHECK-LABEL: @smax125spirv.func @smax(%arg0: i16, %arg1: vector<3xi32>) "None" {126 // CHECK: llvm.intr.smax(%{{.*}}, %{{.*}}) : (i16, i16) -> i16127 %0 = spirv.GL.SMax %arg0, %arg0 : i16128 // CHECK: llvm.intr.smax(%{{.*}}, %{{.*}}) : (vector<3xi32>, vector<3xi32>) -> vector<3xi32>129 %1 = spirv.GL.SMax %arg1, %arg1 : vector<3xi32>130 spirv.Return131}132 133//===----------------------------------------------------------------------===//134// spirv.GL.SMin135//===----------------------------------------------------------------------===//136 137// CHECK-LABEL: @smin138spirv.func @smin(%arg0: i16, %arg1: vector<3xi32>) "None" {139 // CHECK: llvm.intr.smin(%{{.*}}, %{{.*}}) : (i16, i16) -> i16140 %0 = spirv.GL.SMin %arg0, %arg0 : i16141 // CHECK: llvm.intr.smin(%{{.*}}, %{{.*}}) : (vector<3xi32>, vector<3xi32>) -> vector<3xi32>142 %1 = spirv.GL.SMin %arg1, %arg1 : vector<3xi32>143 spirv.Return144}145 146//===----------------------------------------------------------------------===//147// spirv.GL.Sqrt148//===----------------------------------------------------------------------===//149 150// CHECK-LABEL: @sqrt151spirv.func @sqrt(%arg0: f32, %arg1: vector<3xf16>) "None" {152 // CHECK: llvm.intr.sqrt(%{{.*}}) : (f32) -> f32153 %0 = spirv.GL.Sqrt %arg0 : f32154 // CHECK: llvm.intr.sqrt(%{{.*}}) : (vector<3xf16>) -> vector<3xf16>155 %1 = spirv.GL.Sqrt %arg1 : vector<3xf16>156 spirv.Return157}158 159//===----------------------------------------------------------------------===//160// spirv.GL.Tan161//===----------------------------------------------------------------------===//162 163// CHECK-LABEL: @tan164spirv.func @tan(%arg0: f32) "None" {165 // CHECK: llvm.intr.tan(%{{.*}}) : (f32) -> f32166 %0 = spirv.GL.Tan %arg0 : f32167 spirv.Return168}169 170//===----------------------------------------------------------------------===//171// spirv.GL.Tanh172//===----------------------------------------------------------------------===//173 174// CHECK-LABEL: @tanh175spirv.func @tanh(%arg0: f32) "None" {176 // CHECK: llvm.intr.tanh(%{{.*}}) : (f32) -> f32177 %0 = spirv.GL.Tanh %arg0 : f32178 spirv.Return179}180 181//===----------------------------------------------------------------------===//182// spirv.GL.InverseSqrt183//===----------------------------------------------------------------------===//184 185// CHECK-LABEL: @inverse_sqrt186spirv.func @inverse_sqrt(%arg0: f32) "None" {187 // CHECK: %[[ONE:.*]] = llvm.mlir.constant(1.000000e+00 : f32) : f32188 // CHECK: %[[SQRT:.*]] = llvm.intr.sqrt(%{{.*}}) : (f32) -> f32189 // CHECK: llvm.fdiv %[[ONE]], %[[SQRT]] : f32190 %0 = spirv.GL.InverseSqrt %arg0 : f32191 spirv.Return192}193