diff --git a/paddle/fluid/inference/analysis/passes/convert_to_mixed_precision.cc b/paddle/fluid/inference/analysis/passes/convert_to_mixed_precision.cc index 1f7b83504d04e57ac848c862572a8b6213c17f96..e9b188d78f16d970b49bb16cc36c402a636bc68b 100644 --- a/paddle/fluid/inference/analysis/passes/convert_to_mixed_precision.cc +++ b/paddle/fluid/inference/analysis/passes/convert_to_mixed_precision.cc @@ -111,7 +111,7 @@ class ConvertToMixedPrecisionPass { black_list_(black_list), place_(paddle::CPUPlace()), executor_(place_) { - black_list_.insert("assign"); + // black_list_.insert("assign"); black_list_.insert("fill_constant"); black_list_.insert("assign_value"); black_list_.insert("eye"); @@ -416,8 +416,7 @@ void ConvertToMixedPrecisionPass::LoadAndPrepare() { void ConvertToMixedPrecisionPass::FindVarsInMultiBlock() { std::unordered_set all_var_names_set; - std::vector> block_var_names_set( - program_desc_->Size()); + std::vector> block_var_names_set(program_desc_->Size()); for (BlockID idx = 0; idx < program_desc_->Size(); ++idx) { for (auto* op : program_desc_->Block(idx).AllOps()) { const auto& in_names = op->InputArgumentNames(); @@ -449,7 +448,7 @@ void ConvertToMixedPrecisionPass::FindVarsInMultiBlock() { for (const auto& name : vars_in_multi_block) { vars_in_multi_block_with_pair_.emplace( - name, std::make_pair(VarType::FP32, idx)); + name, std::make_pair(VarType::Type(), idx)); } } } @@ -612,7 +611,7 @@ void ConvertToMixedPrecisionPass::ConvertTensorDtype(BlockID block_idx) { if (!has_float_input) { support_precision = false; - VLOG(2) << " op doesn't has float input and output, just skip."; + VLOG(2) << " op doesn't has float input, just skip."; } } VLOG(2) << "op type: " << op_type