blob: 5ee3d73c8a9293b960d9017ff3984a86a9c871a7 (
plain)
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
78
79
80
|
// Copyright (c) 2019 Valve Corporation
// Copyright (c) 2019 LunarG Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#ifndef LIBSPIRV_OPT_RELAX_FLOAT_OPS_PASS_H_
#define LIBSPIRV_OPT_RELAX_FLOAT_OPS_PASS_H_
#include "source/opt/ir_builder.h"
#include "source/opt/pass.h"
namespace spvtools {
namespace opt {
class RelaxFloatOpsPass : public Pass {
public:
RelaxFloatOpsPass() : Pass() {}
~RelaxFloatOpsPass() override = default;
IRContext::Analysis GetPreservedAnalyses() override {
return IRContext::kAnalysisDefUse | IRContext::kAnalysisInstrToBlockMapping;
}
// See optimizer.hpp for pass user documentation.
Status Process() override;
const char* name() const override { return "convert-to-half-pass"; }
private:
// Return true if |inst| can have the RelaxedPrecision decoration applied
// to it.
bool IsRelaxable(Instruction* inst);
// Return true if |inst| returns scalar, vector or matrix type with base
// float and width 32
bool IsFloat32(Instruction* inst);
// Return true if |r_id| is decorated with RelaxedPrecision
bool IsRelaxed(uint32_t r_id);
// If |inst| is an instruction of float32-based type and is not decorated
// RelaxedPrecision, add such a decoration to the module.
bool ProcessInst(Instruction* inst);
// Call ProcessInst on every instruction in |func|.
bool ProcessFunction(Function* func);
Pass::Status ProcessImpl();
// Initialize state for converting to half
void Initialize();
// Set of float result core operations to be processed
std::unordered_set<uint32_t> target_ops_core_f_rslt_;
// Set of float operand core operations to be processed
std::unordered_set<uint32_t> target_ops_core_f_opnd_;
// Set of 450 extension operations to be processed
std::unordered_set<uint32_t> target_ops_450_;
// Set of sample operations
std::unordered_set<uint32_t> sample_ops_;
};
} // namespace opt
} // namespace spvtools
#endif // LIBSPIRV_OPT_RELAX_FLOAT_OPS_PASS_H_
|
