aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/service/compile_only_service.cc
Commit message (Collapse)AuthorAge
* [XLA] Add hook for dump directory expansion.Gravatar Chris Leary2018-09-13
| | | | | | | | | Also puts a ".unoptimized" suffix on dumped HLO protobuf files to avoid the unoptimized dumped HLO protobuf colliding with the optimized dumped HLO protobufs when the same dump directory is specified for both. PiperOrigin-RevId: 212914100
* [XLA] Rename all (Mutable)ArraySlice to absl::Span.Gravatar Tim Shen2018-08-30
| | | | PiperOrigin-RevId: 210998142
* [XLA] Use absl string types and functions instead of the TF versions.Gravatar Justin Lebar2018-08-23
| | | | | | | Unfortunately this has to be one big patch, because e.g. absl::StrCat doesn't accept a TF StringPiece, but as soon as we switch to absl::string_view, we have to switch away from all of the TF functions. PiperOrigin-RevId: 209957896
* Add AotCompilationMetadata field to variant of CompileAheadOfTime.Gravatar Jacques Pienaar2018-06-13
| | | | | | | | | | Add CompileAheadOfTime parameter that can optionally be populated during compilation process. This change is to allow populating metadata even if the CompileAheadOfTime fails. PiperOrigin-RevId: 200407917
* [XLA] Redesign: delete computation_tracker and user_computation.Gravatar A. Unique TensorFlower2018-05-31
| | | | PiperOrigin-RevId: 198743117
* [XLA] Redesign: delete the old service interface.Gravatar A. Unique TensorFlower2018-05-30
| | | | | | | | | | | | | | | | | | - Computation - ComputeConstant - Execute - ExecuteAsync - ExecuteParallel - GetComputationStats - GetComputationShape - GetLocalShape - IsConstant - LoadComputationSnapshot - Op - SetReturnValue - SnapshotComputation PiperOrigin-RevId: 198669035
* [XLA] Redesign: Dump HloSnapshot in local service as well. And support ↵Gravatar A. Unique TensorFlower2018-05-02
| | | | | | replaying HloSnapshot. PiperOrigin-RevId: 195138472
* [XLA] Convert XLA to use xla::se as a namespace alias for ::stream_executor.Gravatar Justin Lebar2018-04-17
| | | | PiperOrigin-RevId: 193301997
* [XLA] Redesign: support xla::XlaComputation in compile-only client and service.Gravatar A. Unique TensorFlower2018-04-17
| | | | PiperOrigin-RevId: 193247845
* [XLA] Implement the whole graph execution interface and make a test use ↵Gravatar A. Unique TensorFlower2018-03-25
| | | | | | | | | | XlaBuilder. - Add Client::ExecuteGraph. - Make client_library_test_base also (partially) support XlaBuilder by using template. - Make one testcase in the axpy_simple_test use XlaBuilder. The test was slightly changed because currently the builder does not expend implicit broadcast automatically. PiperOrigin-RevId: 190268658
* Track DebugOptions in AotCompilationOptionsGravatar Sanjoy Das2018-02-26
| | | | | | In particular, I need this for supporting HLO profiling in the AOT backend. PiperOrigin-RevId: 187081674
* [XLA] Add source mapping utility translation unit, use it in the local client.Gravatar Chris Leary2018-01-25
| | | | PiperOrigin-RevId: 183331075
* [XLA] Add xla_dump_prepass_hlo_proto_to debug flagGravatar Nick Desaulniers2018-01-19
| | | | | | | Add xla_dump_prepass_hlo_proto_to debug flag for dumping HLO protos *BEFORE* running HLO passes, which will perform backend specific optimizations. PiperOrigin-RevId: 182594324
* Remove "hybrid" HloModuleConfig option. The option was used to generate ↵Gravatar Mark Heffernan2017-10-04
| | | | | | | | executables which only generated the array values of tuple-shaped outputs, not the tuple index tables.. With cl/170133015, ShapedBuffers which hold the computation output now have materialized tuples with these index tables so this option is no longer desired or necessary. No functional change. Just cleanup. PiperOrigin-RevId: 171035738
* Create XLA computations inside hostname-specific directory.Gravatar Vijay Vasudevan2017-09-22
| | | | | | | | In a distributed environment, one would like to be able to separate XLA computations based on the binary that produced them, rather than having all computations serialized to a single directory. PiperOrigin-RevId: 169740827
* [TF:XLA] Use HloEvaluator for ComputeConstant, remove the need of a dedicatedGravatar Kay Zhu2017-08-10
| | | | | | compute constant backend. PiperOrigin-RevId: 164940970
* [XLA] Refactor CreateModuleConfig to share code between multiple call-sites.Gravatar Eli Bendersky2017-07-31
| | | | | | | | Previously Service, LocalService and CompileOnlyService had their own code to create a new HloModuleConfig, with much repetition (and some ommissions); collect all these uses in a single method. PiperOrigin-RevId: 163766869
* [XLA] Get rid of ServiceFlags by absorbing it into DebugOptions.Gravatar A. Unique TensorFlower2017-07-18
| | | | | | | After this change HloModuleConfig::hlo_profiling_enabled_ is redundant. I'll remove it in a future change. PiperOrigin-RevId: 162436163
* [XLA] Remove the "hlo dumper" parameter of xla::Compiler and its piping.Gravatar Eli Bendersky2017-06-27
| | | | | | | | | This dumper is no longer necessary since the restructuring of HLO dumping and the addition of MaybeDumpHloModule which heeds to the right flags. The remaining bits didn't have additional functionality, but constituted a lot of boilerplate that has to be propagated throughout the backends. PiperOrigin-RevId: 160281798
* [XLA] Remove dead "in-client" code.Gravatar Mark Heffernan2017-06-21
| | | | | | | | Remove Service::runs_in_client_process_ field and it's dead user. This was previously used by the "InProcess" methods which have been replaced with the LocalClient API. PiperOrigin-RevId: 159759455
* [XLA] Move replica_count out of Backend.Gravatar Eli Bendersky2017-06-20
| | | | | | | | | This is an intermediate step in making replica_count more explicitly programmable (rather than set by a flag). There is no reason for the Backend to hold replica_count - it was only holding it as a container with no additional semantics. PiperOrigin-RevId: 159626528
* [XLA] Replace some XLA CPU compiler specific options by generic "debug options".Gravatar Eli Bendersky2017-06-12
| | | | | | | | LLVM optimization level, extra LLVM flags and "cpu parallel" all turn into debug options on the xla proto. "cpu parallel" is combined with "backend extra options" as a map. PiperOrigin-RevId: 158751784
* Remove unused namespace aliasesGravatar A. Unique TensorFlower2017-05-30
| | | | PiperOrigin-RevId: 157468609
* [XLA] Make HloModule always have a config.Gravatar Eli Bendersky2017-05-24
| | | | | | | | | | | This removes the circular dependency when creating a HloModule, then a HloModuleConfig with the help of the module's entry computation, then assigning the config back to the module. Now we have to pass a config when creating a module, or a default config gets created. This allows removing quite a bit of boilerplate code. PiperOrigin-RevId: 157059949
* [XLA] Attach an HloModuleConfig to HloModule, obviating the need to pass ↵Gravatar Eli Bendersky2017-05-18
| | | | | | | | | | them around as a pair. This cuts through a bunch of critical XLA APIs, but it's time... The background for this change is to make flags/options more easily pipe-able from the TF/XLA boundary deep into the XLA compiler and other components. The situation after this CL is still not perfect; there are a number of places with chicken-egg scenarios when a module has to be constructed before a config (to register the result shape), but the situation is strictly better than before. Future CLs will clean things up even more. PiperOrigin-RevId: 156469639
* Refactor XLA's CompileAheadOfTime out of LocalClient into a new ↵Gravatar A. Unique TensorFlower2017-05-05
CompileOnlyClient class, and likewise from LocalService into a new CompileOnlyService class. This also renames AheadOfTimeComputationInstance to AotComputationInstance for consistency with AotCompilationResult and AotCompilationOptions in compiler/xla/service/compiler.h. Change: 155252320