Commit Graph

  • e60b3eb362 compat: Update forward compatibility horizon to 2025-12-21 A. Unique TensorFlower 2025-12-21 01:03:55 -08:00
  • ff7eb222c2 Refactor XLA Profiler State Check to Use Low-Overhead C API Bhupendra Dubey 2025-12-20 19:50:25 -08:00
  • 580eeae4c3 Automated Code Change A. Unique TensorFlower 2025-12-20 13:49:35 -08:00
  • 9f41228dad Automated Code Change A. Unique TensorFlower 2025-12-20 13:49:31 -08:00
  • 618abe0548 Automated Code Change A. Unique TensorFlower 2025-12-20 13:49:22 -08:00
  • 9799cb0a78 Automated Code Change A. Unique TensorFlower 2025-12-20 13:48:12 -08:00
  • b1b2723815 Automated Code Change A. Unique TensorFlower 2025-12-20 13:47:29 -08:00
  • b1082f97e5 Automated Code Change A. Unique TensorFlower 2025-12-20 13:45:28 -08:00
  • 8633fb9dcf Automated Code Change A. Unique TensorFlower 2025-12-20 13:44:49 -08:00
  • 9e0976de36 Automated Code Change A. Unique TensorFlower 2025-12-20 13:44:46 -08:00
  • 818e61079e Automated Code Change A. Unique TensorFlower 2025-12-20 13:44:45 -08:00
  • 3c431ffe3d Automated Code Change A. Unique TensorFlower 2025-12-20 13:44:41 -08:00
  • 2d5603d02e Automated Code Change A. Unique TensorFlower 2025-12-20 13:42:44 -08:00
  • c0a2b0e8b7 Automated Code Change A. Unique TensorFlower 2025-12-20 13:42:44 -08:00
  • bc145c0cd7 Automated Code Change A. Unique TensorFlower 2025-12-20 13:38:42 -08:00
  • cd1558548f Automated Code Change A. Unique TensorFlower 2025-12-20 13:38:40 -08:00
  • 6e91c3160b Automated Code Change A. Unique TensorFlower 2025-12-20 13:34:17 -08:00
  • 6f24565d89 Automated Code Change A. Unique TensorFlower 2025-12-20 13:27:46 -08:00
  • 68a2b28900 Run third_party/openxla/shardy/google/integrate_latest.sh for the recent unreduced axes. Zixuan Jiang 2025-12-20 13:03:15 -08:00
  • 44a702d0be Reverts 6383e3632c A. Unique TensorFlower 2025-12-20 11:20:21 -08:00
  • 4c4373fe64 Automated Code Change A. Unique TensorFlower 2025-12-20 05:08:08 -08:00
  • e5a57dc540 Automated Code Change A. Unique TensorFlower 2025-12-20 04:36:57 -08:00
  • cd19aba91f Automated Code Change A. Unique TensorFlower 2025-12-20 03:39:28 -08:00
  • f654c43ac2 Automated Code Change A. Unique TensorFlower 2025-12-20 03:06:54 -08:00
  • 39ada7f15c compat: Update forward compatibility horizon to 2025-12-20 A. Unique TensorFlower 2025-12-20 01:04:12 -08:00
  • 9f75c9a8e3 Update GraphDef version to 2447. A. Unique TensorFlower 2025-12-20 01:04:11 -08:00
  • 2c61085107 Automated Code Change A. Unique TensorFlower 2025-12-19 21:54:05 -08:00
  • 269ed39897 Automated Code Change A. Unique TensorFlower 2025-12-19 20:57:56 -08:00
  • 4baf6a3d21 Add replicated -> unreduced test coverage Yash Katariya 2025-12-19 18:06:03 -08:00
  • 8f784b11c3 [XLA][Numerics][HLO Value Tracking] Support HLO original value in CopyFusion pass Jian Cai 2025-12-19 17:55:17 -08:00
  • 9d833374f9 Move mutability tracking from Tracked buffers to RawBuffers as this is only used for importing foreign memory and clutters the API. Parker Schuh 2025-12-19 17:06:23 -08:00
  • f4a923fa82 Add a walk to convert unreduced sdy.constant to replicated sdy.constant and sdy.replicated_to_unreduced. Zixuan Jiang 2025-12-19 17:05:56 -08:00
  • 911ce60c29 Use the XNNPack packing fingerprints to invalidate the weight cache. Quentin Khan 2025-12-19 16:58:16 -08:00
  • 49fda87ad7 Update usage of llvm::PointerType::getUnqual to use non-deprecated version A. Unique TensorFlower 2025-12-19 16:14:19 -08:00
  • cb2c66e715 [XLA:CPU] Enable window dilation support in XLA CPU YNN convolution. Volodymyr Kysenko 2025-12-19 15:55:14 -08:00
  • 3feec3590c Migrate reshape_motion_test to PjRt. Niklas Vangerow 2025-12-19 15:23:06 -08:00
  • af8c7d0e2e Apply llvm-use-new-mlir-op-builder fixes A. Unique TensorFlower 2025-12-19 14:17:57 -08:00
  • 84ad581652 Apply llvm-use-new-mlir-op-builder fixes A. Unique TensorFlower 2025-12-19 14:10:03 -08:00
  • b35e4ed192 [ReplicaGroupV3][Refactor][4/n] Update replica group classes to hide implementation details (via private/protected) to external callers. Bill Varcho 2025-12-19 14:05:31 -08:00
  • d6a407c9f5 Integrate LLVM at llvm/llvm-project@7d381f2a56 A. Unique TensorFlower 2025-12-19 14:02:02 -08:00
  • 5e685fb6e1 Apply llvm-use-new-mlir-op-builder fixes A. Unique TensorFlower 2025-12-19 13:53:16 -08:00
  • d11a803d83 [XLA:GPU] Allow to fuse iotas into sort. Adrian Kuegel 2025-12-19 13:07:58 -08:00
  • e961b23697 [XLA] Continue trying to unroll pipelined loops after failure Vlad Sytchenko 2025-12-19 13:03:01 -08:00
  • 453b5bbfde Internal changes only. Fengwu Yao 2025-12-19 12:54:13 -08:00
  • 07acbd560e [ReplicaGroupV3][Refactor][3/n] Use CollectiveDeviceListBase for polymorphic device list support. Bill Varcho 2025-12-19 12:19:51 -08:00
  • 068c5bfe95 Simplify TrackedCpuDeviceBuffer. Parker Schuh 2025-12-19 11:44:27 -08:00
  • aa95cad5da [XLA][Numerics][HLO Value Tracking] Support HLO original values in ConditionalSimplifier pass Jian Cai 2025-12-19 11:15:41 -08:00
  • c698eecb5a [StableHLO] Update CHLO broadcast ops to use StableHLO bounded-dynamism-aware broadcasting for type inference. Kevin Gleason 2025-12-19 10:54:33 -08:00
  • a3fad5d03e Enable using custom hermetic NCCL version. Yulia Baturina 2025-12-19 10:49:24 -08:00
  • ea362136d9 Allow serialization for sync collectives Maxim Ermilov 2025-12-19 10:43:37 -08:00
  • 843206ca81 Enable using custom hermetic NCCL version. Yulia Baturina 2025-12-19 10:39:43 -08:00
  • 6af9ca2d9c Add LINT.IfChange to keep batch function op registrations in sync. Deqiang Chen 2025-12-19 09:47:30 -08:00
  • 162efc7b6c [PJRT] Add a ExecuteChunk method that allows the user to execute a single chunk of the transpose only. Peter Hawkins 2025-12-19 08:25:36 -08:00
  • 6df9bab4d3 [XLA:GPU] Only mark HoistFusedBitcasts as changed when modifications occur. Theotime Combes 2025-12-19 07:05:39 -08:00
  • 8bb1b4215c PR #35354: [ROCM] bug-fixing SortRewriter and fixing self_adjoint_test on ROCM pemeliya 2025-12-19 05:22:20 -08:00
  • d68f58209e Automated Code Change A. Unique TensorFlower 2025-12-19 02:33:44 -08:00
  • aa1c37b22c Automated Code Change A. Unique TensorFlower 2025-12-19 02:11:35 -08:00
  • bf95b538bf Make filecheck prefixes a parameter. Henning Becker 2025-12-19 02:11:06 -08:00
  • 5ffce0d7c9 [XLA:GPU] Fix bugs when sorting two elements with a fused iota. Adrian Kuegel 2025-12-19 02:10:49 -08:00
  • b9853f94c3 Add more logs to the CuDNN autotuner backend. Dirk Hornung 2025-12-19 01:46:15 -08:00
  • 458fe6aa5a Refactor: Pass GpuComputeCapability to GetBlasComputationType Henning Becker 2025-12-19 01:11:22 -08:00
  • 35dc0c4552 Update GraphDef version to 2446. A. Unique TensorFlower 2025-12-19 01:04:11 -08:00
  • 3d23902120 compat: Update forward compatibility horizon to 2025-12-19 A. Unique TensorFlower 2025-12-19 01:04:04 -08:00
  • 1299abfd8f [XLA:GPU] Add xla_gpu_default_to_alg_dot_bf16_bf16_f32 flag to XLA. Thomas Joerg 2025-12-19 00:53:56 -08:00
  • af68cfc4bd Automated Code Change A. Unique TensorFlower 2025-12-19 00:43:02 -08:00
  • 45c0013db6 Add GetCompilationOptions to PjRt C API Haibo Huang 2025-12-19 00:22:22 -08:00
  • be8b991bd8 Automated Code Change A. Unique TensorFlower 2025-12-19 00:01:55 -08:00
  • 5451b22b40 Remove unused dependencies and conditional compilation from XLA GPU emitter transforms. Henning Becker 2025-12-18 23:43:46 -08:00
  • e8508f09eb Automated Code Change A. Unique TensorFlower 2025-12-18 23:14:10 -08:00
  • c04f83c07d Update symbols in nccl stub for NCCL 2.28.9. Yulia Baturina 2025-12-18 23:07:29 -08:00
  • f1953049fb Fix struct size check in PJRT_Executable_GetCompiledMemoryStats. Haibo Huang 2025-12-18 23:00:15 -08:00
  • 5a8f8d6491 Automated Code Change A. Unique TensorFlower 2025-12-18 20:56:14 -08:00
  • 4b820b349f Automated Code Change A. Unique TensorFlower 2025-12-18 20:19:31 -08:00
  • 72155b4767 Automated Code Change A. Unique TensorFlower 2025-12-18 20:14:18 -08:00
  • 968bb7de37 [PJRT] Remove the .start field on transpose plan Nodes. Peter Hawkins 2025-12-18 18:00:28 -08:00
  • 818d234c94 Add testonly parameter to py_import macros to allow testonly dependencies. Yulia Baturina 2025-12-18 17:39:51 -08:00
  • 46f4c9ab9f [JAX] Refresh a custom layout if a buffer is copied across clients or memories Hyeontaek Lim 2025-12-18 17:11:46 -08:00
  • 728c0ee8e4 Remove unused CoordinationService::Config::recoverable_jobs field. Michael Whittaker 2025-12-18 17:11:14 -08:00
  • c2da241430 Integrate StableHLO at openxla/stablehlo@d496423c Abhinav Gunjal 2025-12-18 16:56:52 -08:00
  • e54cce9f2b (4/N) Add support for NamedSharding in existing HloShardingUtil methods. Remaining methods will be updated in follow up cl's. Kanish Anand 2025-12-18 16:35:41 -08:00
  • 226de359e2 Simplify CoordinationServiceAgent creation. Michael Whittaker 2025-12-18 16:21:42 -08:00
  • decbcf45d9 switch from deprecated TF_CHECK_OK Maxim Ermilov 2025-12-18 15:58:59 -08:00
  • e32237efd3 [XLA:GPU] Split SortRewriter into two passes. Alexander Belyaev 2025-12-18 15:24:36 -08:00
  • 6383e3632c [PJRT] Change transpose code to chunk nodes on the Loop nest representation, not the Node representation. Peter Hawkins 2025-12-18 15:15:38 -08:00
  • 9f210d0ef3 Use a deterministic device assignment for split compilation. Niklas Vangerow 2025-12-18 14:54:51 -08:00
  • 57c1006923 Add ScopedAsyncTrackingEvent to PJRT C API. Haibo Huang 2025-12-18 14:23:01 -08:00
  • a430578a6b Clean up device assignment logic in HloRunnerPjRt. Niklas Vangerow 2025-12-18 14:22:14 -08:00
  • f043b5833d add low_priority queue batch attr to mlrt batch_kernel A. Unique TensorFlower 2025-12-18 14:19:11 -08:00
  • f58771eed6 [XLA] Add DeadDynamicUpdateSliceElimination pass. Cong Liu 2025-12-18 14:01:26 -08:00
  • 3f4f38e2e1 Update bazelisk, buildifier, buildozer and bats-core to the latest version in ml_build Dockerfile Quoc Truong 2025-12-18 13:42:13 -08:00
  • 389c77fa7a Enabling indexing analysis directly on StableHLO. Jacques Pienaar 2025-12-18 13:19:48 -08:00
  • f43a3f21db Expose XProf Collection Walltime Jiya Zhang 2025-12-18 11:48:06 -08:00
  • b4a23c2464 [PJRT C API] Implement xla::PjRtCrossHostSendCancelNotifier. Emily Fertig 2025-12-18 11:43:06 -08:00
  • 6b581e1304 [XLA:CPU] Enable i8xi8->i32 grouped convolutions with ynnpack. Volodymyr Kysenko 2025-12-18 11:40:19 -08:00
  • 5e00c468aa Merge pull request #106221 from tensorflow:fix-bicubic-doc TensorFlower Gardener 2025-12-18 12:12:03 -08:00
  • 649b760aad Add proto serialization for AllToAllStartThunk Maxim Ermilov 2025-12-18 11:08:50 -08:00
  • 38675cad6a Merge pull request #106273 from tensorflow:dependabot/pip/ci/official/requirements_updater/numpy1_requirements/werkzeug-3.1.4 TensorFlower Gardener 2025-12-18 11:53:26 -08:00
  • 73b620f0c2 Remove unneeded CoordinationServiceAgent::ShutdownInternal method. Michael Whittaker 2025-12-18 11:03:55 -08:00
  • c5342da857 Expose execution_options in client_library_test_runner_mixin. Niklas Vangerow 2025-12-18 10:55:21 -08:00
  • e9aa89e2ef Add basic replicated execution support to ClientLibraryTestRunnerMixin<T>. Niklas Vangerow 2025-12-18 10:29:56 -08:00