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