Lines Matching refs:spirv
39 return builder.create<spirv::ConstantOp>(loc, type, in getScalarOrVectorI32Constant()
43 return builder.create<spirv::ConstantOp>(loc, type, in getScalarOrVectorI32Constant()
83 Value signMask = rewriter.create<spirv::ConstantOp>( in matchAndRewrite()
85 Value valueMask = rewriter.create<spirv::ConstantOp>( in matchAndRewrite()
95 rewriter.create<spirv::CompositeConstructOp>(loc, intType, signSplat); in matchAndRewrite()
98 valueMask = rewriter.create<spirv::CompositeConstructOp>(loc, intType, in matchAndRewrite()
103 rewriter.create<spirv::BitcastOp>(loc, intType, adaptor.getLhs()); in matchAndRewrite()
105 rewriter.create<spirv::BitcastOp>(loc, intType, adaptor.getRhs()); in matchAndRewrite()
107 Value value = rewriter.create<spirv::BitwiseAndOp>( in matchAndRewrite()
109 Value sign = rewriter.create<spirv::BitwiseAndOp>( in matchAndRewrite()
112 Value result = rewriter.create<spirv::BitwiseOrOp>(loc, intType, in matchAndRewrite()
114 rewriter.replaceOpWithNewOp<spirv::BitcastOp>(copySignOp, type, result); in matchAndRewrite()
150 Value msb = rewriter.create<spirv::GLFindUMsbOp>(loc, input); in matchAndRewrite()
155 Value subMsb = rewriter.create<spirv::ISubOp>(loc, val31, msb); in matchAndRewrite()
160 Value subInput = rewriter.create<spirv::ISubOp>(loc, val32, input); in matchAndRewrite()
161 Value cmp = rewriter.create<spirv::ULessThanEqualOp>(loc, input, val1); in matchAndRewrite()
162 rewriter.replaceOpWithNewOp<spirv::SelectOp>(countOp, cmp, subInput, in matchAndRewrite()
183 auto one = spirv::ConstantOp::getOne(type, loc, rewriter); in matchAndRewrite()
184 rewriter.replaceOpWithNewOp<spirv::FSubOp>(operation, exp, one); in matchAndRewrite()
203 auto one = spirv::ConstantOp::getOne(type, operation.getLoc(), rewriter); in matchAndRewrite()
205 rewriter.create<spirv::FAddOp>(loc, one, adaptor.getOperand()); in matchAndRewrite()
226 spirv::ConstantOp::getZero(adaptor.getLhs().getType(), loc, rewriter); in matchAndRewrite()
228 rewriter.create<spirv::FOrdLessThanOp>(loc, adaptor.getLhs(), zero); in matchAndRewrite()
229 Value abs = rewriter.create<spirv::GLFAbsOp>(loc, adaptor.getLhs()); in matchAndRewrite()
230 Value pow = rewriter.create<spirv::GLPowOp>(loc, abs, adaptor.getRhs()); in matchAndRewrite()
231 Value negate = rewriter.create<spirv::FNegateOp>(loc, pow); in matchAndRewrite()
232 rewriter.replaceOpWithNewOp<spirv::SelectOp>(powfOp, lessThan, negate, pow); in matchAndRewrite()
249 auto zero = spirv::ConstantOp::getZero(ty, loc, rewriter); in matchAndRewrite()
250 auto one = spirv::ConstantOp::getOne(ty, loc, rewriter); in matchAndRewrite()
253 half = rewriter.create<spirv::ConstantOp>( in matchAndRewrite()
258 half = rewriter.create<spirv::ConstantOp>( in matchAndRewrite()
262 auto abs = rewriter.create<spirv::GLFAbsOp>(loc, operand); in matchAndRewrite()
263 auto floor = rewriter.create<spirv::GLFloorOp>(loc, abs); in matchAndRewrite()
264 auto sub = rewriter.create<spirv::FSubOp>(loc, abs, floor); in matchAndRewrite()
266 rewriter.create<spirv::FOrdGreaterThanEqualOp>(loc, sub, half); in matchAndRewrite()
267 auto select = rewriter.create<spirv::SelectOp>(loc, greater, one, zero); in matchAndRewrite()
268 auto add = rewriter.create<spirv::FAddOp>(loc, floor, select); in matchAndRewrite()
288 .add<CountLeadingZerosPattern, Log1pOpPattern<spirv::GLLogOp>, in populateMathToSPIRVPatterns()
289 ExpM1OpPattern<spirv::GLExpOp>, PowFOpPattern, RoundOpPattern, in populateMathToSPIRVPatterns()
290 spirv::ElementwiseOpPattern<math::AbsOp, spirv::GLFAbsOp>, in populateMathToSPIRVPatterns()
291 spirv::ElementwiseOpPattern<math::CeilOp, spirv::GLCeilOp>, in populateMathToSPIRVPatterns()
292 spirv::ElementwiseOpPattern<math::CosOp, spirv::GLCosOp>, in populateMathToSPIRVPatterns()
293 spirv::ElementwiseOpPattern<math::ExpOp, spirv::GLExpOp>, in populateMathToSPIRVPatterns()
294 spirv::ElementwiseOpPattern<math::FloorOp, spirv::GLFloorOp>, in populateMathToSPIRVPatterns()
295 spirv::ElementwiseOpPattern<math::FmaOp, spirv::GLFmaOp>, in populateMathToSPIRVPatterns()
296 spirv::ElementwiseOpPattern<math::LogOp, spirv::GLLogOp>, in populateMathToSPIRVPatterns()
297 spirv::ElementwiseOpPattern<math::RsqrtOp, spirv::GLInverseSqrtOp>, in populateMathToSPIRVPatterns()
298 spirv::ElementwiseOpPattern<math::SinOp, spirv::GLSinOp>, in populateMathToSPIRVPatterns()
299 spirv::ElementwiseOpPattern<math::SqrtOp, spirv::GLSqrtOp>, in populateMathToSPIRVPatterns()
300 spirv::ElementwiseOpPattern<math::TanhOp, spirv::GLTanhOp>>( in populateMathToSPIRVPatterns()
304 patterns.add<Log1pOpPattern<spirv::CLLogOp>, ExpM1OpPattern<spirv::CLExpOp>, in populateMathToSPIRVPatterns()
305 spirv::ElementwiseOpPattern<math::AbsOp, spirv::CLFAbsOp>, in populateMathToSPIRVPatterns()
306 spirv::ElementwiseOpPattern<math::CeilOp, spirv::CLCeilOp>, in populateMathToSPIRVPatterns()
307 spirv::ElementwiseOpPattern<math::CosOp, spirv::CLCosOp>, in populateMathToSPIRVPatterns()
308 spirv::ElementwiseOpPattern<math::ErfOp, spirv::CLErfOp>, in populateMathToSPIRVPatterns()
309 spirv::ElementwiseOpPattern<math::ExpOp, spirv::CLExpOp>, in populateMathToSPIRVPatterns()
310 spirv::ElementwiseOpPattern<math::FloorOp, spirv::CLFloorOp>, in populateMathToSPIRVPatterns()
311 spirv::ElementwiseOpPattern<math::FmaOp, spirv::CLFmaOp>, in populateMathToSPIRVPatterns()
312 spirv::ElementwiseOpPattern<math::LogOp, spirv::CLLogOp>, in populateMathToSPIRVPatterns()
313 spirv::ElementwiseOpPattern<math::PowFOp, spirv::CLPowOp>, in populateMathToSPIRVPatterns()
314 spirv::ElementwiseOpPattern<math::RoundOp, spirv::CLRoundOp>, in populateMathToSPIRVPatterns()
315 spirv::ElementwiseOpPattern<math::RsqrtOp, spirv::CLRsqrtOp>, in populateMathToSPIRVPatterns()
316 spirv::ElementwiseOpPattern<math::SinOp, spirv::CLSinOp>, in populateMathToSPIRVPatterns()
317 spirv::ElementwiseOpPattern<math::SqrtOp, spirv::CLSqrtOp>, in populateMathToSPIRVPatterns()
318 spirv::ElementwiseOpPattern<math::TanhOp, spirv::CLTanhOp>>( in populateMathToSPIRVPatterns()