Skip to content

Commit

Permalink
test.
Browse files Browse the repository at this point in the history
  • Loading branch information
luweizhou2016 committed Dec 26, 2024
1 parent 852448e commit cf21726
Showing 1 changed file with 5 additions and 5 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ auto available_pred = [](const program_node& input) {
!input.is_type<activation>() && !input.is_type<deconvolution>() && !input.is_type<concatenation>() &&
!input.is_type<crop>() && !input.is_type<eltwise>() && !input.is_type<resample>() &&
!input.is_type<reorder>() && !(input.is_type<permute>() && !input.as<permute>().is_rotating_except_batch()) &&
!input.is_type<strided_slice>())
!input.is_type<strided_slice>() && !input.is_type<reshape>())
return false;
return true;
};
Expand Down Expand Up @@ -158,10 +158,10 @@ bool concat_in_place_optimization::match(const program_node& concat_node,
}
}
// TODO: handle optimized reshape
if (pred.first->is_type<reshape>() && pred.first->can_be_optimized()) {
GPU_DEBUG_TRACE_DETAIL << "[prepare_buffer_fusing_match_fail] : " << pred.first->id() << " , " << concat_node.id() << std::endl;
return false;
}
// if (pred.first->is_type<reshape>() && pred.first->can_be_optimized()) {
// GPU_DEBUG_TRACE_DETAIL << "[prepare_buffer_fusing_match_fail] : " << pred.first->id() << " , " << concat_node.id() << std::endl;
// return false;
// }
// TODO: Investigate if this condition is needed
if (pred.first->get_users().size() > 2) {
GPU_DEBUG_TRACE_DETAIL << "[prepare_buffer_fusing_match_fail] : " << pred.first->id() << " , " << concat_node.id() << std::endl;
Expand Down

0 comments on commit cf21726

Please sign in to comment.