From 891c0d2715cda2a10a49486cf20bfd21faf1f2ce Mon Sep 17 00:00:00 2001 From: Jian Li Date: Wed, 8 Jan 2025 01:11:30 +0000 Subject: [PATCH] Abandon cluster with small size regardless of _XlaCompile attr --- tensorflow/compiler/jit/mark_for_compilation_pass.cc | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tensorflow/compiler/jit/mark_for_compilation_pass.cc b/tensorflow/compiler/jit/mark_for_compilation_pass.cc index 96cf169b914285..37a2aa72d9feb8 100644 --- 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()) {