78 lines · plain
1//==- SPIRVCanonicalization.td - Canonicalization Patterns ---*- tablegen -*==//2 3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8//9// This file defines SPIR-V canonicalization patterns.10//11//===----------------------------------------------------------------------===//12 13include "mlir/IR/PatternBase.td"14include "mlir/Dialect/SPIRV/IR/SPIRVOps.td"15 16//===----------------------------------------------------------------------===//17// spirv.LogicalNot18//===----------------------------------------------------------------------===//19 20def ConvertLogicalNotOfIEqual : Pat<21 (SPIRV_LogicalNotOp (SPIRV_IEqualOp $lhs, $rhs)),22 (SPIRV_INotEqualOp $lhs, $rhs)>;23 24def ConvertLogicalNotOfINotEqual : Pat<25 (SPIRV_LogicalNotOp (SPIRV_INotEqualOp $lhs, $rhs)),26 (SPIRV_IEqualOp $lhs, $rhs)>;27 28def ConvertLogicalNotOfLogicalEqual : Pat<29 (SPIRV_LogicalNotOp (SPIRV_LogicalEqualOp $lhs, $rhs)),30 (SPIRV_LogicalNotEqualOp $lhs, $rhs)>;31 32def ConvertLogicalNotOfLogicalNotEqual : Pat<33 (SPIRV_LogicalNotOp (SPIRV_LogicalNotEqualOp $lhs, $rhs)),34 (SPIRV_LogicalEqualOp $lhs, $rhs)>;35 36//===----------------------------------------------------------------------===//37// spirv.Select -> spirv.GL.*Clamp38//===----------------------------------------------------------------------===//39 40def ValuesAreEqual : Constraint<CPred<"$0 == $1">>;41 42foreach CmpClampPair = [43 [SPIRV_FOrdLessThanOp, SPIRV_GLFClampOp],44 [SPIRV_FOrdLessThanEqualOp, SPIRV_GLFClampOp],45 [SPIRV_SLessThanOp, SPIRV_GLSClampOp],46 [SPIRV_SLessThanEqualOp, SPIRV_GLSClampOp],47 [SPIRV_ULessThanOp, SPIRV_GLUClampOp],48 [SPIRV_ULessThanEqualOp, SPIRV_GLUClampOp]] in {49 50// Detect: $min < $input, $input < $max51def ConvertComparisonIntoClamp1_#CmpClampPair[0] : Pat<52 (SPIRV_SelectOp53 (CmpClampPair[0]54 (SPIRV_SelectOp:$middle055 (CmpClampPair[0] $min, $input),56 $input,57 $min58 ),59 $max60 ),61 $middle1,62 $max),63 (CmpClampPair[1] $input, $min, $max),64 [(ValuesAreEqual $middle0, $middle1)]>;65 66// Detect: $input < $min, $max < $input67def ConvertComparisonIntoClamp2_#CmpClampPair[0] : Pat<68 (SPIRV_SelectOp69 (CmpClampPair[0] $max, $input),70 $max,71 (SPIRV_SelectOp72 (CmpClampPair[0] $input, $min),73 $min,74 $input75 )),76 (CmpClampPair[1] $input, $min, $max)>;77}78