mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2024-12-28 02:31:04 +00:00
5dcb576b69
Motivated by integrating spirv-reduce into spirv-fuzz, so that an added function can be made smaller during shrinking, this adds support in spirv-reduce for asking reduction to be restricted to the instructions of a single specified function.
59 lines
2.1 KiB
C++
59 lines
2.1 KiB
C++
// Copyright (c) 2019 Google LLC
|
|
//
|
|
// 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 SOURCE_REDUCE_REDUCTION_OPPORTUNITY_FINDER_H_
|
|
#define SOURCE_REDUCE_REDUCTION_OPPORTUNITY_FINDER_H_
|
|
|
|
#include <vector>
|
|
|
|
#include "source/opt/ir_context.h"
|
|
#include "source/reduce/reduction_opportunity.h"
|
|
|
|
namespace spvtools {
|
|
namespace reduce {
|
|
|
|
// Abstract class for finding opportunities for reducing a SPIR-V module.
|
|
class ReductionOpportunityFinder {
|
|
public:
|
|
ReductionOpportunityFinder() = default;
|
|
|
|
virtual ~ReductionOpportunityFinder() = default;
|
|
|
|
// Finds and returns the reduction opportunities relevant to this pass that
|
|
// could be applied to SPIR-V module |context|.
|
|
//
|
|
// If |target_function| is non-zero then the available opportunities will be
|
|
// restricted to only those opportunities that modify the function with result
|
|
// id |target_function|.
|
|
virtual std::vector<std::unique_ptr<ReductionOpportunity>>
|
|
GetAvailableOpportunities(opt::IRContext* context,
|
|
uint32_t target_function) const = 0;
|
|
|
|
// Provides a name for the finder.
|
|
virtual std::string GetName() const = 0;
|
|
|
|
protected:
|
|
// Requires that |target_function| is zero or the id of a function in
|
|
// |ir_context|. If |target_function| is zero, returns all the functions in
|
|
// |ir_context|. Otherwise, returns the function with id |target_function|.
|
|
// This allows fuzzer passes to restrict attention to a single function.
|
|
static std::vector<opt::Function*> GetTargetFunctions(
|
|
opt::IRContext* ir_context, uint32_t target_function);
|
|
};
|
|
|
|
} // namespace reduce
|
|
} // namespace spvtools
|
|
|
|
#endif // SOURCE_REDUCE_REDUCTION_OPPORTUNITY_FINDER_H_
|