Lines Matching +full:llvm +full:- +full:3
1 // RUN: mlir-opt -convert-spirv-to-llvm %s | FileCheck %s
3 //===----------------------------------------------------------------------===//
5 //===----------------------------------------------------------------------===//
7 // CHECK-LABEL: @ceil
8 spv.func @ceil(%arg0: f32, %arg1: vector<3xf16>) "None" {
9 // CHECK: "llvm.intr.ceil"(%{{.*}}) : (!llvm.float) -> !llvm.float
11 // CHECK: "llvm.intr.ceil"(%{{.*}}) : (!llvm.vec<3 x half>) -> !llvm.vec<3 x half>
12 %1 = spv.GLSL.Ceil %arg1 : vector<3xf16>
16 //===----------------------------------------------------------------------===//
18 //===----------------------------------------------------------------------===//
20 // CHECK-LABEL: @cos
21 spv.func @cos(%arg0: f32, %arg1: vector<3xf16>) "None" {
22 // CHECK: "llvm.intr.cos"(%{{.*}}) : (!llvm.float) -> !llvm.float
24 // CHECK: "llvm.intr.cos"(%{{.*}}) : (!llvm.vec<3 x half>) -> !llvm.vec<3 x half>
25 %1 = spv.GLSL.Cos %arg1 : vector<3xf16>
29 //===----------------------------------------------------------------------===//
31 //===----------------------------------------------------------------------===//
33 // CHECK-LABEL: @exp
34 spv.func @exp(%arg0: f32, %arg1: vector<3xf16>) "None" {
35 // CHECK: "llvm.intr.exp"(%{{.*}}) : (!llvm.float) -> !llvm.float
37 // CHECK: "llvm.intr.exp"(%{{.*}}) : (!llvm.vec<3 x half>) -> !llvm.vec<3 x half>
38 %1 = spv.GLSL.Exp %arg1 : vector<3xf16>
42 //===----------------------------------------------------------------------===//
44 //===----------------------------------------------------------------------===//
46 // CHECK-LABEL: @fabs
47 spv.func @fabs(%arg0: f32, %arg1: vector<3xf16>) "None" {
48 // CHECK: "llvm.intr.fabs"(%{{.*}}) : (!llvm.float) -> !llvm.float
50 // CHECK: "llvm.intr.fabs"(%{{.*}}) : (!llvm.vec<3 x half>) -> !llvm.vec<3 x half>
51 %1 = spv.GLSL.FAbs %arg1 : vector<3xf16>
55 //===----------------------------------------------------------------------===//
57 //===----------------------------------------------------------------------===//
59 // CHECK-LABEL: @floor
60 spv.func @floor(%arg0: f32, %arg1: vector<3xf16>) "None" {
61 // CHECK: "llvm.intr.floor"(%{{.*}}) : (!llvm.float) -> !llvm.float
63 // CHECK: "llvm.intr.floor"(%{{.*}}) : (!llvm.vec<3 x half>) -> !llvm.vec<3 x half>
64 %1 = spv.GLSL.Floor %arg1 : vector<3xf16>
68 //===----------------------------------------------------------------------===//
70 //===----------------------------------------------------------------------===//
72 // CHECK-LABEL: @fmax
73 spv.func @fmax(%arg0: f32, %arg1: vector<3xf16>) "None" {
74 // CHECK: "llvm.intr.maxnum"(%{{.*}}, %{{.*}}) : (!llvm.float, !llvm.float) -> !llvm.float
76 …// CHECK: "llvm.intr.maxnum"(%{{.*}}, %{{.*}}) : (!llvm.vec<3 x half>, !llvm.vec<3 x half>) -> !ll…
77 %1 = spv.GLSL.FMax %arg1, %arg1 : vector<3xf16>
81 //===----------------------------------------------------------------------===//
83 //===----------------------------------------------------------------------===//
85 // CHECK-LABEL: @fmin
86 spv.func @fmin(%arg0: f32, %arg1: vector<3xf16>) "None" {
87 // CHECK: "llvm.intr.minnum"(%{{.*}}, %{{.*}}) : (!llvm.float, !llvm.float) -> !llvm.float
89 …// CHECK: "llvm.intr.minnum"(%{{.*}}, %{{.*}}) : (!llvm.vec<3 x half>, !llvm.vec<3 x half>) -> !ll…
90 %1 = spv.GLSL.FMin %arg1, %arg1 : vector<3xf16>
94 //===----------------------------------------------------------------------===//
96 //===----------------------------------------------------------------------===//
98 // CHECK-LABEL: @log
99 spv.func @log(%arg0: f32, %arg1: vector<3xf16>) "None" {
100 // CHECK: "llvm.intr.log"(%{{.*}}) : (!llvm.float) -> !llvm.float
102 // CHECK: "llvm.intr.log"(%{{.*}}) : (!llvm.vec<3 x half>) -> !llvm.vec<3 x half>
103 %1 = spv.GLSL.Log %arg1 : vector<3xf16>
107 //===----------------------------------------------------------------------===//
109 //===----------------------------------------------------------------------===//
111 // CHECK-LABEL: @sin
112 spv.func @sin(%arg0: f32, %arg1: vector<3xf16>) "None" {
113 // CHECK: "llvm.intr.sin"(%{{.*}}) : (!llvm.float) -> !llvm.float
115 // CHECK: "llvm.intr.sin"(%{{.*}}) : (!llvm.vec<3 x half>) -> !llvm.vec<3 x half>
116 %1 = spv.GLSL.Sin %arg1 : vector<3xf16>
120 //===----------------------------------------------------------------------===//
122 //===----------------------------------------------------------------------===//
124 // CHECK-LABEL: @smax
125 spv.func @smax(%arg0: i16, %arg1: vector<3xi32>) "None" {
126 // CHECK: "llvm.intr.smax"(%{{.*}}, %{{.*}}) : (!llvm.i16, !llvm.i16) -> !llvm.i16
128 …// CHECK: "llvm.intr.smax"(%{{.*}}, %{{.*}}) : (!llvm.vec<3 x i32>, !llvm.vec<3 x i32>) -> !llvm.v…
129 %1 = spv.GLSL.SMax %arg1, %arg1 : vector<3xi32>
133 //===----------------------------------------------------------------------===//
135 //===----------------------------------------------------------------------===//
137 // CHECK-LABEL: @smin
138 spv.func @smin(%arg0: i16, %arg1: vector<3xi32>) "None" {
139 // CHECK: "llvm.intr.smin"(%{{.*}}, %{{.*}}) : (!llvm.i16, !llvm.i16) -> !llvm.i16
141 …// CHECK: "llvm.intr.smin"(%{{.*}}, %{{.*}}) : (!llvm.vec<3 x i32>, !llvm.vec<3 x i32>) -> !llvm.v…
142 %1 = spv.GLSL.SMin %arg1, %arg1 : vector<3xi32>
146 //===----------------------------------------------------------------------===//
148 //===----------------------------------------------------------------------===//
150 // CHECK-LABEL: @sqrt
151 spv.func @sqrt(%arg0: f32, %arg1: vector<3xf16>) "None" {
152 // CHECK: "llvm.intr.sqrt"(%{{.*}}) : (!llvm.float) -> !llvm.float
154 // CHECK: "llvm.intr.sqrt"(%{{.*}}) : (!llvm.vec<3 x half>) -> !llvm.vec<3 x half>
155 %1 = spv.GLSL.Sqrt %arg1 : vector<3xf16>
159 //===----------------------------------------------------------------------===//
161 //===----------------------------------------------------------------------===//
163 // CHECK-LABEL: @tan
165 // CHECK: %[[SIN:.*]] = "llvm.intr.sin"(%{{.*}}) : (!llvm.float) -> !llvm.float
166 // CHECK: %[[COS:.*]] = "llvm.intr.cos"(%{{.*}}) : (!llvm.float) -> !llvm.float
167 // CHECK: llvm.fdiv %[[SIN]], %[[COS]] : !llvm.float
172 //===----------------------------------------------------------------------===//
174 //===----------------------------------------------------------------------===//
176 // CHECK-LABEL: @tanh
178 // CHECK: %[[TWO:.*]] = llvm.mlir.constant(2.000000e+00 : f32) : !llvm.float
179 // CHECK: %[[X2:.*]] = llvm.fmul %[[TWO]], %{{.*}} : !llvm.float
180 // CHECK: %[[EXP:.*]] = "llvm.intr.exp"(%[[X2]]) : (!llvm.float) -> !llvm.float
181 // CHECK: %[[ONE:.*]] = llvm.mlir.constant(1.000000e+00 : f32) : !llvm.float
182 // CHECK: %[[T0:.*]] = llvm.fsub %[[EXP]], %[[ONE]] : !llvm.float
183 // CHECK: %[[T1:.*]] = llvm.fadd %[[EXP]], %[[ONE]] : !llvm.float
184 // CHECK: llvm.fdiv %[[T0]], %[[T1]] : !llvm.float
189 //===----------------------------------------------------------------------===//
191 //===----------------------------------------------------------------------===//
193 // CHECK-LABEL: @inverse_sqrt
195 // CHECK: %[[ONE:.*]] = llvm.mlir.constant(1.000000e+00 : f32) : !llvm.float
196 // CHECK: %[[SQRT:.*]] = "llvm.intr.sqrt"(%{{.*}}) : (!llvm.float) -> !llvm.float
197 // CHECK: llvm.fdiv %[[ONE]], %[[SQRT]] : !llvm.float