Skip to content

Commit

Permalink
[GPU] Disable crop fusing
Browse files Browse the repository at this point in the history
  • Loading branch information
Lyamin-Roman authored and sshlyapn committed Nov 6, 2024
1 parent 84e6837 commit eaf0947
Showing 1 changed file with 4 additions and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -736,8 +736,6 @@ void prepare_primitive_fusing::fuse_simple_primitives(program &p) {

should_fuse |= input.is_type<strided_slice>();

should_fuse |= input.is_type<crop>();

bool legacy_fusion = activation_node.get_dependencies().size() == 1 &&
!input.can_be_optimized() &&
!activation_node.is_constant() &&
Expand Down Expand Up @@ -922,8 +920,7 @@ void prepare_primitive_fusing::fuse_simple_primitives(program &p) {
(parents[i].first->is_type<gather>()) ||
(parents[i].first->is_type<reduce>() &&
reduce_supports_fusings(parents[i].first->as<reduce>())) ||
(parents[i].first->is_type<lrn>()) ||
(parents[i].first->is_type<crop>());
(parents[i].first->is_type<lrn>());
}

// Disable fusion to a node on constant path when second input is in data flow
Expand Down Expand Up @@ -1063,6 +1060,9 @@ void prepare_primitive_fusing::fuse_simple_primitives(program &p) {
fused_node->get_input_pshape().rbegin()->is_dynamic();

if (is_fc_lora || is_conv_lora || is_gemm_lora) {
if (!can_fuse_parents[peer_idx]) {
return;
}
std::swap(peer_node, fused_node);
}
}
Expand Down

0 comments on commit eaf0947

Please sign in to comment.