Abandon cluster with small size regardless of _XlaCompile attr

ROCm · Jan 8, 2025 · 891c0d2 · 891c0d2
1 parent 8f2d900
commit 891c0d2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tensorflow/compiler/jit/mark_for_compilation_pass.cc b/tensorflow/compiler/jit/mark_for_compilation_pass.cc
@@ -990,8 +990,7 @@ Status MarkForCompilationPassImpl::CreateClusters() {
     // trouble.
 
     if (cluster->effective_cluster_size() >= debug_options_.min_cluster_size ||
-        cluster->has_functional_control_flow() ||
-        cluster->is_xla_compile_attr_true()) {
+        cluster->has_functional_control_flow()) {
       string& name = cluster_names[cluster->cycles_graph_node_id()];
 
       if (name.empty()) {