mirror of
https://github.com/KhronosGroup/SPIRV-Tools
synced 2025-01-15 02:40:04 +00:00
f62357e7b8
This transformation, given the header of a selection construct with branching instruction OpBranchConditional, flattens it. Side-effecting operations such as OpLoad, OpStore and OpFunctionCall are enclosed within smaller conditionals. It is applicable if the construct does not contain inner selection constructs or loops, or atomic or barrier instructions. The corresponding fuzzer pass looks for selection headers and tries to flatten them. Needed for the issue #3544, but it does not fix it completely.
358 lines
20 KiB
C++
358 lines
20 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.
|
|
|
|
#include "source/fuzz/transformation.h"
|
|
|
|
#include <cassert>
|
|
|
|
#include "source/fuzz/fuzzer_util.h"
|
|
#include "source/fuzz/transformation_access_chain.h"
|
|
#include "source/fuzz/transformation_add_constant_boolean.h"
|
|
#include "source/fuzz/transformation_add_constant_composite.h"
|
|
#include "source/fuzz/transformation_add_constant_null.h"
|
|
#include "source/fuzz/transformation_add_constant_scalar.h"
|
|
#include "source/fuzz/transformation_add_copy_memory.h"
|
|
#include "source/fuzz/transformation_add_dead_block.h"
|
|
#include "source/fuzz/transformation_add_dead_break.h"
|
|
#include "source/fuzz/transformation_add_dead_continue.h"
|
|
#include "source/fuzz/transformation_add_function.h"
|
|
#include "source/fuzz/transformation_add_global_undef.h"
|
|
#include "source/fuzz/transformation_add_global_variable.h"
|
|
#include "source/fuzz/transformation_add_image_sample_unused_components.h"
|
|
#include "source/fuzz/transformation_add_local_variable.h"
|
|
#include "source/fuzz/transformation_add_loop_preheader.h"
|
|
#include "source/fuzz/transformation_add_no_contraction_decoration.h"
|
|
#include "source/fuzz/transformation_add_opphi_synonym.h"
|
|
#include "source/fuzz/transformation_add_parameter.h"
|
|
#include "source/fuzz/transformation_add_relaxed_decoration.h"
|
|
#include "source/fuzz/transformation_add_spec_constant_op.h"
|
|
#include "source/fuzz/transformation_add_synonym.h"
|
|
#include "source/fuzz/transformation_add_type_array.h"
|
|
#include "source/fuzz/transformation_add_type_boolean.h"
|
|
#include "source/fuzz/transformation_add_type_float.h"
|
|
#include "source/fuzz/transformation_add_type_function.h"
|
|
#include "source/fuzz/transformation_add_type_int.h"
|
|
#include "source/fuzz/transformation_add_type_matrix.h"
|
|
#include "source/fuzz/transformation_add_type_pointer.h"
|
|
#include "source/fuzz/transformation_add_type_struct.h"
|
|
#include "source/fuzz/transformation_add_type_vector.h"
|
|
#include "source/fuzz/transformation_adjust_branch_weights.h"
|
|
#include "source/fuzz/transformation_composite_construct.h"
|
|
#include "source/fuzz/transformation_composite_extract.h"
|
|
#include "source/fuzz/transformation_composite_insert.h"
|
|
#include "source/fuzz/transformation_compute_data_synonym_fact_closure.h"
|
|
#include "source/fuzz/transformation_duplicate_region_with_selection.h"
|
|
#include "source/fuzz/transformation_equation_instruction.h"
|
|
#include "source/fuzz/transformation_flatten_conditional_branch.h"
|
|
#include "source/fuzz/transformation_function_call.h"
|
|
#include "source/fuzz/transformation_inline_function.h"
|
|
#include "source/fuzz/transformation_invert_comparison_operator.h"
|
|
#include "source/fuzz/transformation_load.h"
|
|
#include "source/fuzz/transformation_make_vector_operation_dynamic.h"
|
|
#include "source/fuzz/transformation_merge_blocks.h"
|
|
#include "source/fuzz/transformation_move_block_down.h"
|
|
#include "source/fuzz/transformation_move_instruction_down.h"
|
|
#include "source/fuzz/transformation_mutate_pointer.h"
|
|
#include "source/fuzz/transformation_outline_function.h"
|
|
#include "source/fuzz/transformation_permute_function_parameters.h"
|
|
#include "source/fuzz/transformation_permute_phi_operands.h"
|
|
#include "source/fuzz/transformation_propagate_instruction_up.h"
|
|
#include "source/fuzz/transformation_push_id_through_variable.h"
|
|
#include "source/fuzz/transformation_record_synonymous_constants.h"
|
|
#include "source/fuzz/transformation_replace_add_sub_mul_with_carrying_extended.h"
|
|
#include "source/fuzz/transformation_replace_boolean_constant_with_constant_binary.h"
|
|
#include "source/fuzz/transformation_replace_constant_with_uniform.h"
|
|
#include "source/fuzz/transformation_replace_copy_memory_with_load_store.h"
|
|
#include "source/fuzz/transformation_replace_copy_object_with_store_load.h"
|
|
#include "source/fuzz/transformation_replace_id_with_synonym.h"
|
|
#include "source/fuzz/transformation_replace_irrelevant_id.h"
|
|
#include "source/fuzz/transformation_replace_linear_algebra_instruction.h"
|
|
#include "source/fuzz/transformation_replace_load_store_with_copy_memory.h"
|
|
#include "source/fuzz/transformation_replace_opphi_id_from_dead_predecessor.h"
|
|
#include "source/fuzz/transformation_replace_opselect_with_conditional_branch.h"
|
|
#include "source/fuzz/transformation_replace_parameter_with_global.h"
|
|
#include "source/fuzz/transformation_replace_params_with_struct.h"
|
|
#include "source/fuzz/transformation_set_function_control.h"
|
|
#include "source/fuzz/transformation_set_loop_control.h"
|
|
#include "source/fuzz/transformation_set_memory_operands_mask.h"
|
|
#include "source/fuzz/transformation_set_selection_control.h"
|
|
#include "source/fuzz/transformation_split_block.h"
|
|
#include "source/fuzz/transformation_store.h"
|
|
#include "source/fuzz/transformation_swap_commutable_operands.h"
|
|
#include "source/fuzz/transformation_swap_conditional_branch_operands.h"
|
|
#include "source/fuzz/transformation_toggle_access_chain_instruction.h"
|
|
#include "source/fuzz/transformation_vector_shuffle.h"
|
|
#include "source/util/make_unique.h"
|
|
|
|
namespace spvtools {
|
|
namespace fuzz {
|
|
|
|
Transformation::~Transformation() = default;
|
|
|
|
std::unique_ptr<Transformation> Transformation::FromMessage(
|
|
const protobufs::Transformation& message) {
|
|
switch (message.transformation_case()) {
|
|
case protobufs::Transformation::TransformationCase::kAccessChain:
|
|
return MakeUnique<TransformationAccessChain>(message.access_chain());
|
|
case protobufs::Transformation::TransformationCase::kAddConstantBoolean:
|
|
return MakeUnique<TransformationAddConstantBoolean>(
|
|
message.add_constant_boolean());
|
|
case protobufs::Transformation::TransformationCase::kAddConstantComposite:
|
|
return MakeUnique<TransformationAddConstantComposite>(
|
|
message.add_constant_composite());
|
|
case protobufs::Transformation::TransformationCase::kAddConstantNull:
|
|
return MakeUnique<TransformationAddConstantNull>(
|
|
message.add_constant_null());
|
|
case protobufs::Transformation::TransformationCase::kAddConstantScalar:
|
|
return MakeUnique<TransformationAddConstantScalar>(
|
|
message.add_constant_scalar());
|
|
case protobufs::Transformation::TransformationCase::kAddCopyMemory:
|
|
return MakeUnique<TransformationAddCopyMemory>(message.add_copy_memory());
|
|
case protobufs::Transformation::TransformationCase::kAddDeadBlock:
|
|
return MakeUnique<TransformationAddDeadBlock>(message.add_dead_block());
|
|
case protobufs::Transformation::TransformationCase::kAddDeadBreak:
|
|
return MakeUnique<TransformationAddDeadBreak>(message.add_dead_break());
|
|
case protobufs::Transformation::TransformationCase::kAddDeadContinue:
|
|
return MakeUnique<TransformationAddDeadContinue>(
|
|
message.add_dead_continue());
|
|
case protobufs::Transformation::TransformationCase::kAddFunction:
|
|
return MakeUnique<TransformationAddFunction>(message.add_function());
|
|
case protobufs::Transformation::TransformationCase::kAddGlobalUndef:
|
|
return MakeUnique<TransformationAddGlobalUndef>(
|
|
message.add_global_undef());
|
|
case protobufs::Transformation::TransformationCase::kAddGlobalVariable:
|
|
return MakeUnique<TransformationAddGlobalVariable>(
|
|
message.add_global_variable());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kAddImageSampleUnusedComponents:
|
|
return MakeUnique<TransformationAddImageSampleUnusedComponents>(
|
|
message.add_image_sample_unused_components());
|
|
case protobufs::Transformation::TransformationCase::kAddLocalVariable:
|
|
return MakeUnique<TransformationAddLocalVariable>(
|
|
message.add_local_variable());
|
|
case protobufs::Transformation::TransformationCase::kAddLoopPreheader:
|
|
return MakeUnique<TransformationAddLoopPreheader>(
|
|
message.add_loop_preheader());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kAddNoContractionDecoration:
|
|
return MakeUnique<TransformationAddNoContractionDecoration>(
|
|
message.add_no_contraction_decoration());
|
|
case protobufs::Transformation::TransformationCase::kAddOpphiSynonym:
|
|
return MakeUnique<TransformationAddOpPhiSynonym>(
|
|
message.add_opphi_synonym());
|
|
case protobufs::Transformation::TransformationCase::kAddParameter:
|
|
return MakeUnique<TransformationAddParameter>(message.add_parameter());
|
|
case protobufs::Transformation::TransformationCase::kAddRelaxedDecoration:
|
|
return MakeUnique<TransformationAddRelaxedDecoration>(
|
|
message.add_relaxed_decoration());
|
|
case protobufs::Transformation::TransformationCase::kAddSpecConstantOp:
|
|
return MakeUnique<TransformationAddSpecConstantOp>(
|
|
message.add_spec_constant_op());
|
|
case protobufs::Transformation::TransformationCase::kAddSynonym:
|
|
return MakeUnique<TransformationAddSynonym>(message.add_synonym());
|
|
case protobufs::Transformation::TransformationCase::kAddTypeArray:
|
|
return MakeUnique<TransformationAddTypeArray>(message.add_type_array());
|
|
case protobufs::Transformation::TransformationCase::kAddTypeBoolean:
|
|
return MakeUnique<TransformationAddTypeBoolean>(
|
|
message.add_type_boolean());
|
|
case protobufs::Transformation::TransformationCase::kAddTypeFloat:
|
|
return MakeUnique<TransformationAddTypeFloat>(message.add_type_float());
|
|
case protobufs::Transformation::TransformationCase::kAddTypeFunction:
|
|
return MakeUnique<TransformationAddTypeFunction>(
|
|
message.add_type_function());
|
|
case protobufs::Transformation::TransformationCase::kAddTypeInt:
|
|
return MakeUnique<TransformationAddTypeInt>(message.add_type_int());
|
|
case protobufs::Transformation::TransformationCase::kAddTypeMatrix:
|
|
return MakeUnique<TransformationAddTypeMatrix>(message.add_type_matrix());
|
|
case protobufs::Transformation::TransformationCase::kAddTypePointer:
|
|
return MakeUnique<TransformationAddTypePointer>(
|
|
message.add_type_pointer());
|
|
case protobufs::Transformation::TransformationCase::kAddTypeStruct:
|
|
return MakeUnique<TransformationAddTypeStruct>(message.add_type_struct());
|
|
case protobufs::Transformation::TransformationCase::kAddTypeVector:
|
|
return MakeUnique<TransformationAddTypeVector>(message.add_type_vector());
|
|
case protobufs::Transformation::TransformationCase::kAdjustBranchWeights:
|
|
return MakeUnique<TransformationAdjustBranchWeights>(
|
|
message.adjust_branch_weights());
|
|
case protobufs::Transformation::TransformationCase::kCompositeConstruct:
|
|
return MakeUnique<TransformationCompositeConstruct>(
|
|
message.composite_construct());
|
|
case protobufs::Transformation::TransformationCase::kCompositeExtract:
|
|
return MakeUnique<TransformationCompositeExtract>(
|
|
message.composite_extract());
|
|
case protobufs::Transformation::TransformationCase::kCompositeInsert:
|
|
return MakeUnique<TransformationCompositeInsert>(
|
|
message.composite_insert());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kComputeDataSynonymFactClosure:
|
|
return MakeUnique<TransformationComputeDataSynonymFactClosure>(
|
|
message.compute_data_synonym_fact_closure());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kDuplicateRegionWithSelection:
|
|
return MakeUnique<TransformationDuplicateRegionWithSelection>(
|
|
message.duplicate_region_with_selection());
|
|
case protobufs::Transformation::TransformationCase::kEquationInstruction:
|
|
return MakeUnique<TransformationEquationInstruction>(
|
|
message.equation_instruction());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kFlattenConditionalBranch:
|
|
return MakeUnique<TransformationFlattenConditionalBranch>(
|
|
message.flatten_conditional_branch());
|
|
case protobufs::Transformation::TransformationCase::kFunctionCall:
|
|
return MakeUnique<TransformationFunctionCall>(message.function_call());
|
|
case protobufs::Transformation::TransformationCase::kInlineFunction:
|
|
return MakeUnique<TransformationInlineFunction>(
|
|
message.inline_function());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kInvertComparisonOperator:
|
|
return MakeUnique<TransformationInvertComparisonOperator>(
|
|
message.invert_comparison_operator());
|
|
case protobufs::Transformation::TransformationCase::kLoad:
|
|
return MakeUnique<TransformationLoad>(message.load());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kMakeVectorOperationDynamic:
|
|
return MakeUnique<TransformationMakeVectorOperationDynamic>(
|
|
message.make_vector_operation_dynamic());
|
|
case protobufs::Transformation::TransformationCase::kMergeBlocks:
|
|
return MakeUnique<TransformationMergeBlocks>(message.merge_blocks());
|
|
case protobufs::Transformation::TransformationCase::kMoveBlockDown:
|
|
return MakeUnique<TransformationMoveBlockDown>(message.move_block_down());
|
|
case protobufs::Transformation::TransformationCase::kMoveInstructionDown:
|
|
return MakeUnique<TransformationMoveInstructionDown>(
|
|
message.move_instruction_down());
|
|
case protobufs::Transformation::TransformationCase::kMutatePointer:
|
|
return MakeUnique<TransformationMutatePointer>(message.mutate_pointer());
|
|
case protobufs::Transformation::TransformationCase::kOutlineFunction:
|
|
return MakeUnique<TransformationOutlineFunction>(
|
|
message.outline_function());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kPermuteFunctionParameters:
|
|
return MakeUnique<TransformationPermuteFunctionParameters>(
|
|
message.permute_function_parameters());
|
|
case protobufs::Transformation::TransformationCase::kPermutePhiOperands:
|
|
return MakeUnique<TransformationPermutePhiOperands>(
|
|
message.permute_phi_operands());
|
|
case protobufs::Transformation::TransformationCase::kPropagateInstructionUp:
|
|
return MakeUnique<TransformationPropagateInstructionUp>(
|
|
message.propagate_instruction_up());
|
|
case protobufs::Transformation::TransformationCase::kPushIdThroughVariable:
|
|
return MakeUnique<TransformationPushIdThroughVariable>(
|
|
message.push_id_through_variable());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kRecordSynonymousConstants:
|
|
return MakeUnique<TransformationRecordSynonymousConstants>(
|
|
message.record_synonymous_constants());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceAddSubMulWithCarryingExtended:
|
|
return MakeUnique<TransformationReplaceAddSubMulWithCarryingExtended>(
|
|
message.replace_add_sub_mul_with_carrying_extended());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceBooleanConstantWithConstantBinary:
|
|
return MakeUnique<TransformationReplaceBooleanConstantWithConstantBinary>(
|
|
message.replace_boolean_constant_with_constant_binary());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceConstantWithUniform:
|
|
return MakeUnique<TransformationReplaceConstantWithUniform>(
|
|
message.replace_constant_with_uniform());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceCopyMemoryWithLoadStore:
|
|
return MakeUnique<TransformationReplaceCopyMemoryWithLoadStore>(
|
|
message.replace_copy_memory_with_load_store());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceCopyObjectWithStoreLoad:
|
|
return MakeUnique<TransformationReplaceCopyObjectWithStoreLoad>(
|
|
message.replace_copy_object_with_store_load());
|
|
case protobufs::Transformation::TransformationCase::kReplaceIdWithSynonym:
|
|
return MakeUnique<TransformationReplaceIdWithSynonym>(
|
|
message.replace_id_with_synonym());
|
|
case protobufs::Transformation::TransformationCase::kReplaceIrrelevantId:
|
|
return MakeUnique<TransformationReplaceIrrelevantId>(
|
|
message.replace_irrelevant_id());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceLinearAlgebraInstruction:
|
|
return MakeUnique<TransformationReplaceLinearAlgebraInstruction>(
|
|
message.replace_linear_algebra_instruction());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceLoadStoreWithCopyMemory:
|
|
return MakeUnique<TransformationReplaceLoadStoreWithCopyMemory>(
|
|
message.replace_load_store_with_copy_memory());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceOpselectWithConditionalBranch:
|
|
return MakeUnique<TransformationReplaceOpSelectWithConditionalBranch>(
|
|
message.replace_opselect_with_conditional_branch());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceParameterWithGlobal:
|
|
return MakeUnique<TransformationReplaceParameterWithGlobal>(
|
|
message.replace_parameter_with_global());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceParamsWithStruct:
|
|
return MakeUnique<TransformationReplaceParamsWithStruct>(
|
|
message.replace_params_with_struct());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kReplaceOpphiIdFromDeadPredecessor:
|
|
return MakeUnique<TransformationReplaceOpPhiIdFromDeadPredecessor>(
|
|
message.replace_opphi_id_from_dead_predecessor());
|
|
case protobufs::Transformation::TransformationCase::kSetFunctionControl:
|
|
return MakeUnique<TransformationSetFunctionControl>(
|
|
message.set_function_control());
|
|
case protobufs::Transformation::TransformationCase::kSetLoopControl:
|
|
return MakeUnique<TransformationSetLoopControl>(
|
|
message.set_loop_control());
|
|
case protobufs::Transformation::TransformationCase::kSetMemoryOperandsMask:
|
|
return MakeUnique<TransformationSetMemoryOperandsMask>(
|
|
message.set_memory_operands_mask());
|
|
case protobufs::Transformation::TransformationCase::kSetSelectionControl:
|
|
return MakeUnique<TransformationSetSelectionControl>(
|
|
message.set_selection_control());
|
|
case protobufs::Transformation::TransformationCase::kSplitBlock:
|
|
return MakeUnique<TransformationSplitBlock>(message.split_block());
|
|
case protobufs::Transformation::TransformationCase::kStore:
|
|
return MakeUnique<TransformationStore>(message.store());
|
|
case protobufs::Transformation::TransformationCase::kSwapCommutableOperands:
|
|
return MakeUnique<TransformationSwapCommutableOperands>(
|
|
message.swap_commutable_operands());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kSwapConditionalBranchOperands:
|
|
return MakeUnique<TransformationSwapConditionalBranchOperands>(
|
|
message.swap_conditional_branch_operands());
|
|
case protobufs::Transformation::TransformationCase::
|
|
kToggleAccessChainInstruction:
|
|
return MakeUnique<TransformationToggleAccessChainInstruction>(
|
|
message.toggle_access_chain_instruction());
|
|
case protobufs::Transformation::TransformationCase::kVectorShuffle:
|
|
return MakeUnique<TransformationVectorShuffle>(message.vector_shuffle());
|
|
case protobufs::Transformation::TRANSFORMATION_NOT_SET:
|
|
assert(false && "An unset transformation was encountered.");
|
|
return nullptr;
|
|
}
|
|
assert(false && "Should be unreachable as all cases must be handled above.");
|
|
return nullptr;
|
|
}
|
|
|
|
bool Transformation::CheckIdIsFreshAndNotUsedByThisTransformation(
|
|
uint32_t id, opt::IRContext* ir_context,
|
|
std::set<uint32_t>* ids_used_by_this_transformation) {
|
|
if (!fuzzerutil::IsFreshId(ir_context, id)) {
|
|
return false;
|
|
}
|
|
if (ids_used_by_this_transformation->count(id) != 0) {
|
|
return false;
|
|
}
|
|
ids_used_by_this_transformation->insert(id);
|
|
return true;
|
|
}
|
|
|
|
} // namespace fuzz
|
|
} // namespace spvtools
|