1 // Copyright (c) 2020 André Perez Maselco
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 //     http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 
15 #ifndef SOURCE_FUZZ_TRANSFORMATION_ADJUST_BRANCH_WEIGHTS_H_
16 #define SOURCE_FUZZ_TRANSFORMATION_ADJUST_BRANCH_WEIGHTS_H_
17 
18 #include "source/fuzz/protobufs/spirvfuzz_protobufs.h"
19 #include "source/fuzz/transformation.h"
20 #include "source/fuzz/transformation_context.h"
21 #include "source/opt/ir_context.h"
22 
23 namespace spvtools {
24 namespace fuzz {
25 
26 class TransformationAdjustBranchWeights : public Transformation {
27  public:
28   explicit TransformationAdjustBranchWeights(
29       const protobufs::TransformationAdjustBranchWeights& message);
30 
31   TransformationAdjustBranchWeights(
32       const protobufs::InstructionDescriptor& instruction_descriptor,
33       const std::pair<uint32_t, uint32_t>& branch_weights);
34 
35   // - |message_.instruction_descriptor| must identify an existing
36   //   branch conditional instruction
37   // - At least one of |branch_weights| must be non-zero and
38   //   the two weights must not overflow a 32-bit unsigned integer when added
39   //   together
40   bool IsApplicable(
41       opt::IRContext* ir_context,
42       const TransformationContext& transformation_context) const override;
43 
44   // Adjust the branch weights of a branch conditional instruction.
45   void Apply(opt::IRContext* ir_context,
46              TransformationContext* transformation_context) const override;
47 
48   std::unordered_set<uint32_t> GetFreshIds() const override;
49 
50   protobufs::Transformation ToMessage() const override;
51 
52  private:
53   protobufs::TransformationAdjustBranchWeights message_;
54 };
55 
56 }  // namespace fuzz
57 }  // namespace spvtools
58 
59 #endif  // SOURCE_FUZZ_TRANSFORMATION_ADJUST_BRANCH_WEIGHTS_H_
60