experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Print values for parsed MarkForCompilationPassFlags and XlaOpsCommonFlags

Browse Source 
While at it change from "typedef struct { ... } X;" to "struct X { ... };".  The
former is a C-ism not necessary in C++.

PiperOrigin-RevId: 221089377
This commit is contained in:
Sanjoy Das 2018-11-12 07:35:16 -08:00 committed by TensorFlower Gardener
parent d8ba8138a2
commit 499b8bd16d
3 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/compiler/jit/legacy_flags/mark_for_compilation_pass_flags.cc
View File

@ -20,6 +20,7 @@ limitations under the License.
#include "tensorflow/compiler/jit/legacy_flags/mark_for_compilation_pass_flags.h"
#include "tensorflow/compiler/xla/parse_flags_from_env.h"
#include "tensorflow/core/platform/logging.h"
#include "tensorflow/core/platform/types.h"
#include "tensorflow/core/util/command_line_flags.h"
@ -65,6 +66,17 @@ static void AllocateFlags() {
"enable fusion of element-wise operations only using XLA when "
"global_jit_level is ON*.")});
xla::ParseFlagsFromEnv(*flag_list);
if (VLOG_IS_ON(1)) {
VLOG(1) << "Parsed MarkForCompilationPassFlags:";
VLOG(1) << " tf_xla_auto_jit = " << flags->tf_xla_auto_jit;
VLOG(1) << " tf_xla_min_cluster_size = " << flags->tf_xla_min_cluster_size;
VLOG(1) << " tf_xla_max_cluster_size = " << flags->tf_xla_max_cluster_size;
VLOG(1) << " tf_xla_clustering_debug = " << flags->tf_xla_clustering_debug;
VLOG(1) << " tf_xla_cpu_global_jit = " << flags->tf_xla_cpu_global_jit;
VLOG(1) << " tf_xla_clustering_fuel = " << flags->tf_xla_clustering_fuel;
VLOG(1) << " tf_xla_fusion_only = " << flags->tf_xla_fusion_only;
}
}
// Append to *append_to flag definitions associated with the XLA bridge's

4
tensorflow/compiler/jit/legacy_flags/mark_for_compilation_pass_flags.h
View File

@ -33,7 +33,7 @@ void AppendMarkForCompilationPassFlags(
// The values of flags associated with the XLA bridge's
// mark_for_compilation_pass module.
typedef struct {
struct MarkForCompilationPassFlags {
int32 tf_xla_auto_jit; // Control compilation of operators into XLA
// computations on CPU and GPU devices. 0 = use
// ConfigProto setting; -1 = off; 1 = on for things
@ -55,7 +55,7 @@ typedef struct {
// is set to ON* and overrides its behavior. If
// true, enable fusion of element-wise operations
// only using XLA.
} MarkForCompilationPassFlags;
};
// Return a pointer to the MarkForCompilationPassFlags struct;
// repeated calls return the same pointer.

8
tensorflow/compiler/jit/legacy_flags/xla_ops_common_flags.cc
View File

@ -18,7 +18,7 @@ limitations under the License.
#include "tensorflow/compiler/jit/legacy_flags/xla_ops_common_flags.h"
#include "tensorflow/compiler/xla/parse_flags_from_env.h"
#include "tensorflow/core/platform/logging.h"
#include "tensorflow/core/util/command_line_flags.h"
namespace tensorflow {
@ -36,6 +36,12 @@ void AllocateAndParseFlags() {
&flags->tf_xla_always_defer_compilation, ""),
});
xla::ParseFlagsFromEnv(*flag_list);
if (VLOG_IS_ON(1)) {
VLOG(1) << "Parsed XlaOpsCommonFlags:";
VLOG(1) << " tf_xla_always_defer_compilation = "
<< flags->tf_xla_always_defer_compilation;
}
}
const XlaOpsCommonFlags& GetXlaOpsCommonFlags() {