[dGPU] avoid strided_slice to executed in cpu

riverlijunjie · Jul 17, 2024 · f933ddf · f933ddf
1 parent b8ba903
commit f933ddf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/plugins/intel_gpu/src/graph/graph_optimizer/mark_shape_of_subgraphs.cpp b/src/plugins/intel_gpu/src/graph/graph_optimizer/mark_shape_of_subgraphs.cpp
@@ -7,6 +7,7 @@
 #include "reshape_inst.h"
 #include "eltwise_inst.h"
 #include "select_inst.h"
+#include "strided_slice_inst.h"
 #include "gather_inst.h"
 #include "pass_manager.h"
 
@@ -107,7 +108,7 @@ void mark_shape_of_subgraphs::mark_node(program_node& node) {
     // Update impl if needed
     const auto default_subgraph_impl = impl_types::cpu;
     if (_update_impls)
-        if (!node.is_type<reshape>())
+        if (!node.is_type<reshape>() && !node.is_type<strided_slice>())
             node.set_preferred_impl_type(default_subgraph_impl);
 }
 

diff --git a/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp b/src/plugins/intel_gpu/src/graph/layout_optimizer.cpp
@@ -1516,7 +1516,7 @@ impl_types layout_optimizer::get_preferred_impl_type(program_node& node, format
     if (forced_impl != impl_types::any)
         return forced_impl;
 
-    if (node.is_in_shape_of_subgraph() && !node.is_type<reshape>())
+    if (node.is_in_shape_of_subgraph() && !node.is_type<reshape>() && !node.is_type<strided_slice>())
         return impl_types::cpu;
 
     if (!_forcing_map.empty() && _forcing_map.count(node.id()) != 0) {