| Commit message (Collapse) | Author | Age |
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
| |
PiperOrigin-RevId: 210998142
|
|
|
|
|
|
|
| |
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 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
|
|
|
|
| |
PiperOrigin-RevId: 198743117
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
- Computation
- ComputeConstant
- Execute
- ExecuteAsync
- ExecuteParallel
- GetComputationStats
- GetComputationShape
- GetLocalShape
- IsConstant
- LoadComputationSnapshot
- Op
- SetReturnValue
- SnapshotComputation
PiperOrigin-RevId: 198669035
|
|
|
|
|
|
| |
replaying HloSnapshot.
PiperOrigin-RevId: 195138472
|
|
|
|
| |
PiperOrigin-RevId: 193301997
|
|
|
|
| |
PiperOrigin-RevId: 193247845
|
|
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
|
|
| |
In particular, I need this for supporting HLO profiling in the AOT backend.
PiperOrigin-RevId: 187081674
|
|
|
|
| |
PiperOrigin-RevId: 183331075
|
|
|
|
|
|
|
| |
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
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
|
|
| |
compute constant backend.
PiperOrigin-RevId: 164940970
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
|
|
|
| |
After this change HloModuleConfig::hlo_profiling_enabled_ is redundant. I'll
remove it in a future change.
PiperOrigin-RevId: 162436163
|
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
| |
PiperOrigin-RevId: 157468609
|
|
|
|
|
|
|
|
|
|
|
| |
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
|
|
|
|
|
|
|
|
|
|
| |
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
|
|
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
|