OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 00:19:58 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity
186,511 Commits 50 Branches 144 Tags 1.3 GiB
master
Commit Graph

186511 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
A. Unique TensorFlower
1d634b4b7a Update GraphDef version to 2395. 
PiperOrigin-RevId: 825439806
 2025-10-29 02:27:43 -07:00
A. Unique TensorFlower
801e45d17f compat: Update forward compatibility horizon to 2025-10-29 
PiperOrigin-RevId: 825439792
 2025-10-29 02:13:51 -07:00
Will Froom
a8356f63df [XLA:CPU] Fix UnrollExtractLoops to stop false-positive unroll when the vector itself is dependent on for loop. 
PiperOrigin-RevId: 825427562
 2025-10-29 01:57:31 -07:00
Alexander Belyaev
f19413ab39 [XLA:GPU] Move SymbolicExpr out of the gpu:: namespace. 
PiperOrigin-RevId: 825420017
 2025-10-29 01:43:55 -07:00
Kanish Anand
fb796c3475 Clarify maximal mesh documentation. 
PiperOrigin-RevId: 825417349
 2025-10-29 01:34:30 -07:00
A. Unique TensorFlower
12f2b7d4e6 Automated Code Change 
PiperOrigin-RevId: 825413784
 2025-10-29 01:24:25 -07:00
A. Unique TensorFlower
364a0ac855 [XLA:GPU] Add simple reduce check to multimem tests 
PiperOrigin-RevId: 825413069
 2025-10-29 01:00:17 -07:00
A. Unique TensorFlower
c0405a3895 Automated Code Change 
PiperOrigin-RevId: 825375527
 2025-10-28 23:01:20 -07:00
Vlad Sytchenko
db42f74169 Googly changes four words 
PiperOrigin-RevId: 825371929
 2025-10-28 22:47:58 -07:00
Eugene Zhulenev
27a45465a0 [xla:codegen] Remove MlirKernelEmitter alias 
PiperOrigin-RevId: 825346639
 2025-10-28 21:20:29 -07:00
Eugene Zhulenev
1444679887 [xla:codegen] Remove LlvmKernelDefinition alias 
Rely on KernelEmitter<T>::KernelDefinition alias and CTAD to avoid spelling full type.

PiperOrigin-RevId: 825336093
 2025-10-28 20:48:11 -07:00
Eugene Zhulenev
8e76d82f01 [xla:codegen] Remove LlvmKernelEmitter alias 
PiperOrigin-RevId: 825312592
 2025-10-28 20:34:13 -07:00
Eugene Zhulenev
f2ecdb670b [xla:ffi] Make XLA_FFI_TypeInfo a versioned struct with size and extension 
PiperOrigin-RevId: 825311188
 2025-10-28 20:25:38 -07:00
Peter Hawkins
6efea71d95 Update nanobind to commit e507b118927bc3a12446d0ca235e1baaf343932e. 
This includes https://github.com/wjakob/nanobind/pull/1191, which fixes a data race observed in free threading builds.

PiperOrigin-RevId: 825304073
 2025-10-28 20:15:37 -07:00
Subhankar Shah
a7fb82d82d [XLA:MSA] Allow block prefetching for values that have aliased uses - 
* Extend alternate memory chunk reservations for aliased uses.
* Add pinned allocations in alternate memory for aliased uses.
* Mark all aliased allocations as colocated.

Pin all values aliased to the left of the source hlo value being block prefetched to default memory, values aliased to the right of the source hlo value will be pinned to default memory in all cases except - when a scheduling a new prefetch is successful.
* For source values, if a pinned allocation exists and adding uses to the allocation is required.
* For other (aliased) values finalizing the value suffices.

Misc:
* When creating colocated aliased allocations in alternate memory, do not rely on getting the first colocated block from the back of the list.
* Code cleanup.
PiperOrigin-RevId: 825302368
 2025-10-28 20:06:21 -07:00
A. Unique TensorFlower
075912cbd6 Do not legalize mhlo.iota to tf.range when element type is ui8, since tf.range doesn't support ui8. 
PiperOrigin-RevId: 825298715
 2025-10-28 19:56:25 -07:00
Vlad Sytchenko
de3ce6ee98 [XLA] Allow fine grain SparseCore offloading 
PiperOrigin-RevId: 825298200
 2025-10-28 19:43:00 -07:00
Eugene Zhulenev
6ef78e9c38 [xla:codegen] Parametrize KernelEmitter by kernel source type 
Remove one level of template indirection by always parametrizing by a kernel source type.

PiperOrigin-RevId: 825295806
 2025-10-28 19:36:37 -07:00
Alexander Shaposhnikov
b81ecb432f Add initial bits to support reductions. 
PiperOrigin-RevId: 825292462
 2025-10-28 19:08:17 -07:00
Zixuan Jiang
e3549cef96 Refactor spmd/shardy export_named_computations and import_func_calls. 
PiperOrigin-RevId: 825290126
 2025-10-28 18:57:37 -07:00
David Majnemer
9b433c3f5a Remove deprecated Shape(const ShapeProto&) constructor. 
This change removes the deprecated `Shape(const ShapeProto&)` constructor and updates its call sites to use `Shape::FromProto` instead, which returns a `StatusOr<Shape>`. The call sites now explicitly handle the potential error status.

PiperOrigin-RevId: 825288664
 2025-10-28 18:44:36 -07:00
Eugene Zhulenev
1a17862b09 [xla:codegen] Fix warnings in KernelDefinition + cleanup API a bit 
PiperOrigin-RevId: 825283848
 2025-10-28 18:38:12 -07:00
Niklas Vangerow
c49788eb0a Migrate fft_test to use PjRt. 
PiperOrigin-RevId: 825262773
 2025-10-28 18:19:53 -07:00
Subhankar Shah
a7654beb5c Cleanup: Remove types and _types modules from xla/python/tools because it is dead code 
PiperOrigin-RevId: 825262464
 2025-10-28 18:11:11 -07:00
Bryan Massoth
f5f9bd8099 Upgrade TraceMeProducer/Consumer traces as kCritical due to them being required to group traces across async boundaries. 
PiperOrigin-RevId: 825245546
 2025-10-28 17:50:08 -07:00
Jian Cai
8e130fe8be [XLA][Numerics][HLO Value Tracking] Print the entire instruction for mismatch tuple structures 
This prints the entire instruction instead of instruction name when the original value has a different tuple structure with the instruction shape in HLO verifier pass. The instruction name is not reliable and could be different from its name in the HLO dump.

PiperOrigin-RevId: 825241833
 2025-10-28 17:30:30 -07:00
Eugene Zhulenev
a72cd9ceeb [xla:codegen] Move KernelEmitter::name() to base class and change type to absl::string_view 
PiperOrigin-RevId: 825237762
 2025-10-28 17:21:00 -07:00
Kevin Gleason
03f4c66dd1 [StableHLO Optim] Add CompareOp patterns and dont fold large converts. 
PiperOrigin-RevId: 825236477
 2025-10-28 17:11:15 -07:00
Peter Gavin
1e0c214c2f Internal build rule change. 
PiperOrigin-RevId: 825236018
 2025-10-28 16:57:42 -07:00
A. Unique TensorFlower
f8f2123d5a Add another call to ynn_optimize_subgraph 
PiperOrigin-RevId: 825233680
 2025-10-28 16:47:05 -07:00
Karlo Basioli
8ab9ebee94 [XLA:CPU] Guard LLVM command line options when infering target machine 
PiperOrigin-RevId: 825233596
 2025-10-28 16:33:16 -07:00
A. Unique TensorFlower
281fa6f4d3 Fix dot library predicates 
This changes the predicates for calling a library for a dot to indicate whether we will actually call the library, not just whether the library supports the dot. This fixes bugs where we incorrectly claim a library will handle the dot.

PiperOrigin-RevId: 825231317
 2025-10-28 16:23:53 -07:00
Parker Schuh
d7b371034b Update users not to set untuple_result now that it is true by default. 
PiperOrigin-RevId: 825230517
 2025-10-28 16:15:35 -07:00
Eugene Zhulenev
3ea8166731 [xla:ffi] Add xla::ffi::MakeTypeInfo 
PiperOrigin-RevId: 825230372
 2025-10-28 16:05:25 -07:00
Eugene Zhulenev
1d829c0f02 [xla] Update documentation to use xla::Future 
PiperOrigin-RevId: 825222421
 2025-10-28 15:56:02 -07:00
A. Unique TensorFlower
24abec31c7 Add 1D array test case for SPMD DUS (both to verify the HLO transformation and the overall correctness) 
PiperOrigin-RevId: 825222307
 2025-10-28 15:46:59 -07:00
Jian Cai
66078903f7 [XLA][Numerics][HLO Original Value] Support original values for some cases in while loop simplifier pass 
This updates the original value of a while loop if any unused parameters are removed.

PiperOrigin-RevId: 825221785
 2025-10-28 15:34:04 -07:00
Majid Dadashi
2feb74eeff Add i4 support in tfl.slice 
PiperOrigin-RevId: 825217744
 2025-10-28 15:27:41 -07:00
Haibo Huang
202bd1ac59 Remove IsTpuTopology and IsGpuTopology and IsCpuTopology 
PiperOrigin-RevId: 825216555
 2025-10-28 15:17:52 -07:00
Matt Hurd
803a513588 Allow specifying a custom session id instead of always using timestamp. 
This change introduces a `session_id` option in `ProfileOptions` and `RemoteProfilerSessionManagerOptions`. When provided, this id will be used as the subdirectory for storing profile data, instead of the default timestamp.

PiperOrigin-RevId: 825216123
 2025-10-28 15:10:15 -07:00
Parker Schuh
fef8806609 Remove unnecessary limit. 
PiperOrigin-RevId: 825212681
 2025-10-28 14:54:58 -07:00
Antonio Sanchez
b531e70088 Updating internal visibility rules. 
PiperOrigin-RevId: 825202091
 2025-10-28 14:26:07 -07:00
Victor Stone
63a9d0d1f8 If device placement annotations are found inside host computations (as a result of nested host computations), hoist them up the call stack. If any unsupported cases or inconsistencies are detected, an error will be returned to the user. 
This allows JAX's migration from their previous `compute_on` API to the new (currently named `compute_on2`) API.

PiperOrigin-RevId: 825177029
 2025-10-28 13:24:57 -07:00
A. Unique TensorFlower
768e653c9c Integrate LLVM at llvm/llvm-project@29c830cbf8 
Updates LLVM usage to match
[29c830cbf8c6](https://github.com/llvm/llvm-project/commit/29c830cbf8c6)

PiperOrigin-RevId: 825166312
 2025-10-28 13:03:12 -07:00
Peter Gavin
5c13ee5063 Internal build rule change. 
PiperOrigin-RevId: 825151475
 2025-10-28 12:22:01 -07:00
A. Unique TensorFlower
98c995ee9a Remove unnecessary native.bind calls from workspace0.bzl. 
These bindings for gRPC, Python headers, and six are no longer required.

PiperOrigin-RevId: 825146137
 2025-10-28 12:09:32 -07:00
Eugene Zhulenev
44446df9cf [xla:cpu] Remove mlir/llvm kernel_definition and kernel_emitter libraries 
PiperOrigin-RevId: 825133247
 2025-10-28 11:46:01 -07:00
Bill Varcho
63d558e46f [ReplicaGroupV3][Mesh + AxesRef] add to/from proto functions + equality op to XLA definitions of Mesh and AxesRef. 
PiperOrigin-RevId: 825107889
 2025-10-28 11:18:27 -07:00
Henning Becker
4dfbd3bd0c Add proto serialization for RocmComputeCapability 
PiperOrigin-RevId: 825103988
 2025-10-28 11:08:48 -07:00
Eugene Zhulenev
fa61547732 [xla:cpu] Rename LlvmIrKernelSource to LlvmKernelSource 
PiperOrigin-RevId: 825103892
 2025-10-28 10:59:06 -07:00