diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-05-24 14:08:30 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-05-24 14:11:54 -0700 |
| commit | 58d5281515d2c1873ca94541b15f6114ab20554a (patch) | |
| tree | 88e466629b0e1c068e53d58a92c5b59977495c44 | |
| parent | 3be6124bcc7b41c94cad3f96fb62ec423abfd110 (diff) | |
Add various XLA flags to tfcompile.
PiperOrigin-RevId: 157034590
| -rw-r--r-- | tensorflow/compiler/aot/BUILD | 7 | ||||
| -rw-r--r-- | tensorflow/compiler/aot/tfcompile_main.cc | 14 |
2 files changed, 21 insertions, 0 deletions
diff --git a/tensorflow/compiler/aot/BUILD b/tensorflow/compiler/aot/BUILD index c12005a4ca..5e368749a0 100644 --- a/tensorflow/compiler/aot/BUILD +++ b/tensorflow/compiler/aot/BUILD @@ -123,9 +123,16 @@ cc_library( deps = [ ":tfcompile_lib", ":tfcompile_proto", + "//tensorflow/compiler/xla/legacy_flags:alias_analysis_flags", + "//tensorflow/compiler/xla/legacy_flags:buffer_assignment_flags", "//tensorflow/compiler/xla/legacy_flags:compiler_functor_flags", "//tensorflow/compiler/xla/legacy_flags:cpu_compiler_flags", "//tensorflow/compiler/xla/legacy_flags:cpu_runtime_flags", + "//tensorflow/compiler/xla/legacy_flags:hlo_graph_dumper_flags", + "//tensorflow/compiler/xla/legacy_flags:hlo_pass_pipeline_flags", + "//tensorflow/compiler/xla/legacy_flags:llvm_util_flags", + "//tensorflow/compiler/xla/legacy_flags:service_flags", + "//tensorflow/compiler/xla/legacy_flags:util_flags", "//tensorflow/compiler/xla/service:compiler", "//tensorflow/core:core_cpu", "//tensorflow/core:core_cpu_internal", diff --git a/tensorflow/compiler/aot/tfcompile_main.cc b/tensorflow/compiler/aot/tfcompile_main.cc index 4b7e220769..63ec261e01 100644 --- a/tensorflow/compiler/aot/tfcompile_main.cc +++ b/tensorflow/compiler/aot/tfcompile_main.cc @@ -23,9 +23,16 @@ limitations under the License. #include "tensorflow/compiler/aot/flags.h" #include "tensorflow/compiler/aot/tfcompile.pb.h" #include "tensorflow/compiler/aot/tfcompile_util.h" +#include "tensorflow/compiler/xla/legacy_flags/alias_analysis_flags.h" +#include "tensorflow/compiler/xla/legacy_flags/buffer_assignment_flags.h" #include "tensorflow/compiler/xla/legacy_flags/compiler_functor_flags.h" #include "tensorflow/compiler/xla/legacy_flags/cpu_compiler_flags.h" #include "tensorflow/compiler/xla/legacy_flags/cpu_runtime_flags.h" +#include "tensorflow/compiler/xla/legacy_flags/hlo_graph_dumper_flags.h" +#include "tensorflow/compiler/xla/legacy_flags/hlo_pass_pipeline_flags.h" +#include "tensorflow/compiler/xla/legacy_flags/llvm_util_flags.h" +#include "tensorflow/compiler/xla/legacy_flags/service_flags.h" +#include "tensorflow/compiler/xla/legacy_flags/util_flags.h" #include "tensorflow/compiler/xla/service/compiler.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/graph.pb.h" @@ -129,9 +136,16 @@ int main(int argc, char** argv) { std::vector<tensorflow::Flag> flag_list; AppendMainFlags(&flag_list, &flags); + xla::legacy_flags::AppendAliasAnalysisFlags(&flag_list); + xla::legacy_flags::AppendBufferAssignmentFlags(&flag_list); xla::legacy_flags::AppendCompilerFunctorFlags(&flag_list); xla::legacy_flags::AppendCpuCompilerFlags(&flag_list); xla::legacy_flags::AppendCpuRuntimeFlags(&flag_list); + xla::legacy_flags::AppendHloGraphDumperFlags(&flag_list); + xla::legacy_flags::AppendHloPassPipelineFlags(&flag_list); + xla::legacy_flags::AppendLlvmUtilFlags(&flag_list); + xla::legacy_flags::AppendServiceFlags(&flag_list); + xla::legacy_flags::AppendUtilFlags(&flag_list); tensorflow::string usage = tensorflow::tfcompile::kUsageHeader; usage += tensorflow::Flags::Usage(argv[0], flag_list); |