225 Pull requests merged by 4 people

• [IFRT] Add simple string array tests to IFRT common array_impl_test

  #92823 merged May 7, 2025

• [ReplicaGroupV2] Generate all-to-all collectives using IotaReplicaGroupList format.

  #92755 merged May 7, 2025

• Add a bool flag to UpdateCompileOptions to specify whether the addressable device lookup is required for compilation.

  #92747 merged May 7, 2025

• Use ArrayRef instead of tsl::RCReference<Array>

  #92898 merged May 7, 2025

• Clarify the difference between internal and external version.

  #92830 merged May 7, 2025

• A simple constant folder for case ops

  #92812 merged May 7, 2025

• Add strings::LegacyPrecision to TSL.

  #92890 merged May 7, 2025

• Force disable build time Python pyc generation.

  #92905 merged May 7, 2025

• This CL has no functional changes, but improved code structure:

  #92091 merged May 7, 2025

• Adding an option to initialize Pluggable Device Plugin using function pointers instead of a DSO

  #92816 merged May 7, 2025

• Fix missing cuda graph launch call back event in new version of CUDA

  #92904 merged May 7, 2025

• Refactor BFS function to accept distance parameter

  #92444 merged May 7, 2025

• Branch quantization/stablehlo/cc:post_calibration to use TFQuantDialect

  #92889 merged May 7, 2025

• Introduce new CompilePhaseHloRunnerPjRt and ExecutePhaseHloRunnerPjRt.

  #92440 merged May 7, 2025

• Branch quantization/tensorflow:quantize_preprocess to use TFQuantDialect

  #92886 merged May 7, 2025

• LogScheduleStatistics is very slow. Lower it's level from 1 to 3.

  #89198 merged May 7, 2025

• [XLA:benchmarks] Add --workflow_type to the binary to generate GHA input matrices by workflow types

  #92903 merged May 7, 2025

• Fix HloRunnerPjRt DeserializeExecutable.

  #92902 merged May 7, 2025

• Migration fuse_convolution_pass to tf_attrs_and_constraints

  #92558 merged May 7, 2025

• Refactor the DiffSummary log print.

  #92628 merged May 7, 2025

• Branch quantization/stablehlo/cc:tf_pass_pipeline to use TFQuantDialect

  #92432 merged May 7, 2025

• Add an API to the URL generator for generating URLs without selecting nodes.

  #92725 merged May 7, 2025

• [IFRT] Treat Maximal HloSharding on a single device as fully replicated

  #92810 merged May 7, 2025

• Clean up LoadedExecutable::Delete and LoadedExecutable::IsDeleted

  #92791 merged May 7, 2025

• Remove support for sparse dimension's unique attribute.

  #92753 merged May 7, 2025

• Support memory space in underlying buffer allocation.

  #92821 merged May 7, 2025

• Add an optimization pattern to convert a fully_connected op with a weight with last dimension of 1 to a broadcasting mul op.

  #92893 merged May 7, 2025

• Support types that have AbslStringify, but not operator<< in the additional output of TF_RET_CHECK.

  #92720 merged May 7, 2025

• [JAX] Add a test for multiprocess shard_map in McJAX with non-participating hosts. Update handling of device memories in PyDeviceList to support this.

  #92737 merged May 7, 2025

• [XLA:benchmarks] Add a binary to generate GHA input matrices to trigger benchmark runs

  #92807 merged May 7, 2025

• Remove Shape::ShortDebugString() to reduce the Shape API surface.

  #92565 merged May 7, 2025

• Only delegate 4 bit weights which are per channel quantized

  #92891 merged May 7, 2025

• [XLA:GPU] Refactor RedzoneBuffers. Make pararameters of RedzoneBuffers explicit and decouple RedzoneBuffers from AutotunerCompileUtil and AutotunerCompileConfig.

  #92878 merged May 7, 2025

• [StableHLO] Fix typo.

  #92789 merged May 7, 2025

• [xla::ifrt] Rename xla::ifrt::Compiler::Compile to xla::ifrt::Compiler::CompileAndLoad and point xla::ifrt::Compiler::Compile to xla::ifrt::Compiler::CompileAndLoad.

  #92506 merged May 7, 2025

• Reverts eb4629ef35970666a34683d13d7fc8c5b421118a

  #92883 merged May 7, 2025

• #sdy Apply xla.sdy.FuncResultSharding if it has no uses or it's used by an op that isn't func.return.

  #92882 merged May 7, 2025

• [XLA:GPU] Simplify gpu_hlo_schedule_test using absl::ScopedMockLog.

  #92787 merged May 7, 2025

• Remove obsolete TODO

  #92799 merged May 7, 2025

• [XLA:GPU] Favor pipelined over synchronous collectives when combining.

  #92795 merged May 7, 2025

• [XLA:GPU] (NFC) Refactor ShapeToMlirTypes to use ForEachLeafShape.

  #92784 merged May 7, 2025

• [xla:cpu] Remove Tfrt from PJRT CPU buffer type names

  #92758 merged May 7, 2025

• [XLA:GPU] (NFC) Simplify logic in CreateMLIRModule.

  #92783 merged May 7, 2025

• PR #25919: feat: add runtime_version field to target_config

  #92711 merged May 7, 2025

• Record manual axes instead of generating mhlo.sharding in stablehlo/shard_map_export.

  #92533 merged May 7, 2025

• Automated Code Change

  #92863 merged May 7, 2025

• Automated Code Change

  #92857 merged May 7, 2025

• [XLA:GPU][Emitters] Correctly handle allocate_shared for s4 type.

  #92879 merged May 7, 2025

• Automated Code Change

  #92860 merged May 7, 2025

• [XLA:CPU] Migrate xla_jit_compiled_cpu_function to thunk execution

  #90379 merged May 7, 2025

• PR #25441: [XLA:GPU] add cuDNN Paged attention support

  #92703 merged May 7, 2025

• #sdy no need to add (and remove) a fake result to a callback custom call with no results.

  #92794 merged May 7, 2025

• PR #26101: Allow bitcasts in dynamic memcpy fusions.

  #92785 merged May 7, 2025

• PR #26111: [NFC] Refactor ResolveFunctionalDependencyOnInductionVariable.

  #92866 merged May 7, 2025

• [XLA:GPU] Add simple triton support test for tuple ops

  #91496 merged May 7, 2025

• [XLA:CPU] Expose interface for rendering execution graphs via dot API

  #92618 merged May 7, 2025

• Rollback oneDNN v3.7 + ACL v24.12 update in aarch64 build.

  #92809 merged May 7, 2025

• Use shared_ptr instead of unique_ptr for executable references

  #92826 merged May 7, 2025

• PR #25959: [ROCm] fixed missing headers + deps

  #92702 merged May 7, 2025

• [xla] Add a name to execution graph operations

  #92687 merged May 7, 2025

• Temporarily add UBDialect as a dependent dialect in SdyRoundTripStablehloToHloToStablehloPass.

  #92872 merged May 7, 2025

• [IFRT] Exclude sdy dialect from the verification conducted by the IFRT SPMD expandable verification pass

  #92770 merged May 7, 2025

• [XLA:GPU] Integrate perf tables to SoL estimator.

  #92412 merged May 7, 2025

• Bump the Shardy XLA pin.

  #92869 merged May 7, 2025

• Introduce a type alias for array references

  #92800 merged May 7, 2025

• Automated Code Change

  #92862 merged May 7, 2025

• fix(WeakTrackableReference): correct weak reference type handling

  #92336 merged May 7, 2025

• Temporarily roll back changes for new LLVM version

  #92828 merged May 7, 2025

• Added support for callback with LayoutModesToXlaShapes when auto-sharding modifies shapes of arguments or results.

  #92763 merged May 7, 2025

• Fork remaining passes to tf_passes and related .mlir tests in with new namespace

  #92541 merged May 7, 2025

• Use LoadedExecutableRef instead of std::unique_ptr<LoadedExecutable>

  #92815 merged May 6, 2025

• Replace usage of scipy.misc.derivative in TensorFlow.

  #92822 merged May 6, 2025

• Add two optimization patterns replacing x + (y - y) or (y - y) + x with x.

  #92811 merged May 6, 2025

• Remove tf/compiler/mlir/quantization/common/quantization_lib and tf/compiler/mlir/quantization/tensorflow/ops:tf_op_quant_spec from tf_passes

  #92730 merged May 6, 2025

• Integrate StableHLO at openxla/stablehlo@63246aea

  #92446 merged May 6, 2025

• [XLA] Remove UB dialect from XLA dependencies.

  #92722 merged May 6, 2025

• [xla:gpu] Detect empty traced CUDA graphs

  #92806 merged May 6, 2025

• Add Fprint128ToBytes utility function.

  #92749 merged May 6, 2025

• Reverts 38a0e6bdaea46322fbeffb0902a8136fe8bc9130

  #92803 merged May 6, 2025

• Remove tf/compiler/mlir/lite:tensorflow_lite from tf/compiler/mlir/quantization/tensorflow:passes

  #92416 merged May 6, 2025

• Move BufferAllocationSliceProto to xla/service/buffer_assignment.proto

  #92786 merged May 6, 2025

• Precompute set of buffers allocated/released per instruction.

  #92739 merged May 6, 2025

• Add swizzle_mode to TMA attribute

  #92790 merged May 6, 2025

• [xla:ffi] Fix typo in struct size used during DeviceOrdinal decoding.

  #92797 merged May 6, 2025

• PR #26100: Clean up special cases for AsyncStart in HloComputation.

  #92780 merged May 6, 2025

• Add a bazel config setting to indicate whether the code is built in a CI.

  #92516 merged May 6, 2025

• Introduce aliases for executable references

  #92792 merged May 6, 2025

• PR #22788: Implement GetCompiledMemoryStats for GPU AOT executables

  #92796 merged May 6, 2025

• Integrate support for tensordesc structs and reenable descriptor tests

  #92411 merged May 6, 2025

• Add proto serialization for the SequentialThunk

  #92772 merged May 6, 2025

• [XLA:GPU] Add CuBLAS autotuner backend.

  #92269 merged May 6, 2025

• Rewrite the argument filtering code to use argument masks

  #92410 merged May 6, 2025

• [XLA:GPU]: Remove GpuPerformanceModelOptions

  #92717 merged May 6, 2025

• #sdy Handle the case where an output of a sdy.manual_computation has a dimension of size zero.

  #92781 merged May 6, 2025

• Remove unused dependency on flatbuffer_import

  #92728 merged May 6, 2025

• [XLA:GPU] Add support for .txtpb files.

  #92743 merged May 6, 2025

• [XLA:GPU] Remove unused trace argument from CreateMLIRModule.

  #92779 merged May 6, 2025

• Drop unneeded dep from dialect

  #92606 merged May 6, 2025

• PR #26046: Generalize dynamic memcpy support.

  #92713 merged May 6, 2025

• Automated Code Change

  #92691 merged May 6, 2025

• [xla:gpu][triton] Update TMA lowering in XLA to use new api

  #92716 merged May 6, 2025

• [XLA:GPU] Collect matmul perf tables on forge.

  #92712 merged May 6, 2025

• [XLA] Disable dynamic-slice async conversion until we correctly account for operands.

  #92757 merged May 6, 2025

• Automated Code Change

  #92647 merged May 6, 2025

• Remove unused header and dep.

  #92707 merged May 6, 2025

• [XLA:GPU] Remove xla_gpu_unsafe_pipelined_loop_annotator flag

  #89920 merged May 6, 2025

• [XLA:GPU] Remove special case for send/recv in async events mapping

  #92740 merged May 6, 2025

• Fix missing dep.

  #92727 merged May 6, 2025

• [XLA:GPU] Remove brittle pipeline parallelism detection in the SPMD partitioner pass

  #90065 merged May 6, 2025

• Support platform / memory space allocator in TfrtGpuClient.

  #92750 merged May 6, 2025

• [xla:pjrt] Add micro benchmarks for StreamExecutorGpuClient

  #92738 merged May 6, 2025

• Initialize fp16_to_fp32 with identity map

  #92633 merged May 6, 2025

• [XLA:GPU] Move all_reduce_splitter to collectives folder

  #92511 merged May 6, 2025

• TransposeConv bias should be matching the dimension of the output channel, and should always be 1D if exists. Flat the succeeding bias constant before fusion.

  #92689 merged May 6, 2025

• Add tfl.EmbeddingLookup DRQ

  #92742 merged May 6, 2025

• [XLA:GPU] Move all_reduce_decomposer to collectives folder

  #92512 merged May 6, 2025

• [XLA:GPU] Move all_reduce_blueconnect to collectives folder

  #92509 merged May 5, 2025

• Updates IOPDDL conversion to shift edge costs by a constant amount if negative values are present.

  #92726 merged May 5, 2025

• Implement TransferToInfeed and TransferFromOutfeed in pjrt async gpu client.

  #92588 merged May 5, 2025

• Internal only changes

  #92733 merged May 5, 2025

• [XLA:GPU] Move all_gather_dynamic_slice_simplifier to collectives folder

  #92510 merged May 5, 2025

• Automated Code Change

  #92719 merged May 5, 2025

• [XLA:benchmarks] Update benchmark proto and add implementations to load and parse proto for triggering benchmark run using GitHub Actions

  #92619 merged May 5, 2025

• [XLA:GPU] Move all_gather_optimizer to collectives folder

  #92508 merged May 5, 2025

• Refactor HloRunnerPjRt to distinguish loaded and unloaded PjRtExecutables.

  #92630 merged May 5, 2025

• [IFRT] Switch std::shared_ptr<xla::ifrt::Sharding> to xla::ifrt::ShardingRef

  #92627 merged May 5, 2025

• Prepare VarHandle ops too

  #92721 merged May 5, 2025

• Avoid host scheduling dependencies between two GPU CUDA requests

  #91890 merged May 5, 2025

• [JAX] In McJAX, support JIT compilation/execution on only the devices attached to a single host, with other hosts not participating.

  #91849 merged May 5, 2025

• Implement Auotuner Backend interface for Triton.

  #92710 merged May 5, 2025

• [XLA:GPU/TMA] Modified some TMA tests based on investigations in their failures. Incorrect results were caused by incorrect handling of swizzling modes with regards to TMA. After the latest Triton integration, swizzling is determined during the compilation pipeline and should be extracted from TTGIR. This will be addressed independently.

  #92718 merged May 5, 2025

• [XLA:GPU] update autotune db entry

  #92715 merged May 5, 2025

• [XLA:GPU] Add simple triton support test for map op

  #91655 merged May 5, 2025

• Automated Code Change

  #92645 merged May 5, 2025

• Remove more if_gpu_is_configured usages from service/gpu/BUILD

  #92486 merged May 5, 2025

• [XLA:GPU] Check whether the propagated tile offsets can be used.

  #90805 merged May 5, 2025

• PR #25953: [XLA:FFI][BugFix] Add Token Args and Rets for typed FFI on CPU backend

  #92709 merged May 5, 2025

• PR #25916: Added SafeStaticInit for deadlock free safe static initialization

  #92708 merged May 5, 2025

• PR #25965: [NVIDIA GPU] Fix fabric info test

  #92607 merged May 5, 2025

• PR #26003: Run fewer analyses in HloEvaluator.

  #92616 merged May 5, 2025

• PR #26001: [NVIDIA GPU] Change UB register cache to include size in the key

  #92705 merged May 5, 2025

• PR #24279: Handle FP8 bitcasting when lowering tensors

  #92611 merged May 5, 2025

• PR #26012: Convert computations to MakeNonfusionComputations in async wrappers

  #92704 merged May 5, 2025

• [XLA:Python] [JAX] Move HloPass bindings to XLA/Python.

  #92688 merged May 5, 2025

• Automated Code Change

  #92676 merged May 4, 2025

• [xla:pjrt] Fix deadlock in tracked device buffer

  #92661 merged May 3, 2025

• Automated Code Change

  #92648 merged May 3, 2025

• Add a clarifying comment in sparse_core_layout.proto

  #92566 merged May 3, 2025

• Automated Code Change

  #92325 merged May 3, 2025

• Automated Code Change

  #92601 merged May 3, 2025

• Automated Code Change

  #92599 merged May 3, 2025

• Change TfrtCpuClient to inherit from CommonPjRtClient and implement the

  #92585 merged May 3, 2025

• [xla] Deprecate CustomCallApiVersion::API_VERSION_ORIGINAL

  #92617 merged May 3, 2025

• fix clang tidy linter checks

  #92631 merged May 3, 2025

• Add cpu implementations of CommonPjRtRawBuffer, PjRtDeviceEvent, and

  #92525 merged May 3, 2025

• Use HloSliceInstruction directly instead of HloInstruction.

  #92070 merged May 3, 2025

• Refactor HloRunnerPjRt to use absl_nonnull annotation.

  #92632 merged May 3, 2025

• Update libTPU versions.

  #92623 merged May 2, 2025

• [xla:pjrt] Hide CpuDeviceMemory implementations in .cc file

  #92621 merged May 2, 2025

• Fix linter warnings and remove unnecessary const.

  #92161 merged May 2, 2025

• [XLA:GPU] Deprecate xla_gpu_unsafe_pipelined_loop_annotator

  #89919 merged May 2, 2025

• [xla:pjrt] AsyncValueRef is thread safe and no need to guard it with a mutex

  #92586 merged May 2, 2025

• [IFRT] Introduce xla::ifrt::ShardingRef type alias

  #92582 merged May 2, 2025

• When there are addressable devices for the given device assignment, use an addressable device for compilation. Otherwise, use device 0 as the default compilation device.

  #92612 merged May 2, 2025

• Propagate context to waiters correctly. We sometimes resume execution from a shared thread (particularly in the TPU code) and currently the context is lost to all following operations.

  #92490 merged May 2, 2025

• CreateToken, Send, Recv, GetTupleElement, Infeed, Outfeed : Direct StableHLO to HLO Translation

  #92063 merged May 2, 2025

• Disable a few broken PJRT unittests because the feature they are testing is simply not implemented at all

  #92620 merged May 2, 2025

• move forked tf_passes to tf_quant namespace and use tf_quant_td_files for generated files

  #92449 merged May 2, 2025

• Added Shutdown method to PreemptionSyncManager.

  #91600 merged May 2, 2025

• Fixed concurrent collective creation bug.

  #92574 merged May 2, 2025

• Fix typos in documentation strings

  #92584 merged May 2, 2025

• [CI] Update twine version in ml-build docker files

  #92614 merged May 2, 2025

• Simplify ProgramShape API by removing ShortDebugString() and DebugString().

  #92562 merged May 2, 2025

• Add the ability to specify log severity to TF_RET_CHECK.

  #92561 merged May 2, 2025

• Don't add unnecessary dependencies.

  #92570 merged May 2, 2025

• Update ManualComputation round-tripping to not have round-tripping attrs on the CallOps.

  #91381 merged May 2, 2025

• Clean up status macro implementations:

  #92583 merged May 2, 2025

• Remove Shape::SerializeAsString() to reduce the Shape API surface.

  #92567 merged May 2, 2025

• Remove obsolete comment

  #92589 merged May 2, 2025

• Automated Code Change

  #92532 merged May 2, 2025

• [xla:pjrt] PjRtFuture: refresh documentation and rename unique to is_move_only

  #92604 merged May 2, 2025

• Bump ubuntu from ed1544e to d80997d in /tensorflow/tools/tf_sig_build_dockerfiles

  #92538 merged May 2, 2025

• Added options into the profiler.

  #92498 merged May 2, 2025

• Automated Code Change

  #92454 merged May 2, 2025

• Bump ubuntu from 8e5c4f0 to 8feb4d8 in /ci/official/containers/linux_arm64

  #92542 merged May 2, 2025

• Bump ubuntu from 7229784 to 1e622c5 in /tensorflow/tools/gcs_test

  #92539 merged May 2, 2025

• Automated Code Change

  #92543 merged May 2, 2025

• Bump the github-actions group with 3 updates

  #92540 merged May 2, 2025

• Automated Code Change

  #92324 merged May 2, 2025

• Fix potential conflict on the newly assigned nvtx line id for NVTX CUPTI events. Put all new line id assignment into the post_processing_single_host_xplane.cc.

  #92435 merged May 2, 2025

• Return error when copy data to pinned host memory space is not implemented.

  #92580 merged May 2, 2025

• Reduce cache misses in LatencyHidingScheduler.

  #92554 merged May 2, 2025

• Make it clear that accessing a Shape's properties incompatible with its kind is a programmer error. The new error messages are more detailed and more actionable.

  #92557 merged May 2, 2025

• Fix the failure in the edge case when there are no mirror tars for the specified CUDA version.

  #92579 merged May 2, 2025

• Add int2 dtypes to TensorFlow.

  #92262 merged May 1, 2025

• #sdy mark xla.sdy.PropagationBarrier as side effecting so it doesn't disappear or get deduped during CSE.

  #92577 merged May 1, 2025

• Fix bug where HloConstantInstruction::PrintOperandsWithCanonicalNameMap() access dimensions() before ensuring that the shape is an array.

  #92560 merged May 1, 2025

• PR #25926: [GPU] Guard se_gpu_pjrt_client tests not expected to pass in OSS.

  #92537 merged May 1, 2025

• [IFRT] Introduce xla::ifrt::ShardingRef type alias

  #92571 merged May 1, 2025

• PR #25707: [XLA:GPU] Fix cuDNN command buffer incorrect update

  #92578 merged May 1, 2025

• Remove Shape::SetProto(), which is redundant with Shape::ToProto().

  #92568 merged May 1, 2025

• [TFLite] Fix incorrect FuseMulAndFullyConnected optimization

  #92569 merged May 1, 2025

• Fix crash in TfrtGpuBuffer::DropExternalReference().

  #92526 merged May 1, 2025

• [xla:gpu] Use object pool of FFI CallFrames in custom call thunk and command

  #92524 merged May 1, 2025

• Adding absl_py dependency for pip_package test_pip_package

  #92234 merged May 1, 2025

• Integrate LLVM at llvm/llvm-project@7752e0a10b25

  #92517 merged May 1, 2025

• Use the default hash function for node_to_context_.

  #92555 merged May 1, 2025

• Add vlog messages in CudaExecutor

  #92559 merged May 1, 2025

• fork preprocess_op.cc to tf_preprocess_op.cc without lite dependencies

  #92012 merged May 1, 2025

• [xla] ObjectPool: add memory order to all atomic operations

  #92552 merged May 1, 2025

• Don't reference the wrapped unconstructed buffer if it's in error state.

  #92419 merged May 1, 2025

• Branch quantization/stablehlo/instrumentations:save_report to use TFQuantDialect

  #92507 merged May 1, 2025

• Replaced WorkerThread with SingleThreadedExecutor.

  #92495 merged May 1, 2025

• Reverts 6aad90114b5a78885b3087741992ffc496a79127

  #92284 merged May 1, 2025

• [XLA:CPU][roll forward] The breakage wasn't caused by this CL

  #92501 merged May 1, 2025

• [XLA:CPU][tfcompile] Serialize tfcompile required thunks into execute invocations

  #92494 merged May 1, 2025

• Record and check for client type when serialize / deserialize Executable.

  #92452 merged May 1, 2025

• PR #22814: Enable BF16 detection on Aarch64

  #92492 merged May 1, 2025

• [XLA:CPU] Add utility to recurse into a thunk sequence proto

  #92491 merged May 1, 2025

• Add BufferFromHostLiteral into CommonPjRtClient.

  #92244 merged May 1, 2025

• [xla] Fix data race in ObjectPool

  #92514 merged May 1, 2025

• Fix: Use the prerelease label for Python 3.14.0a6 rules_python url path

  #92513 merged May 1, 2025

• Branch quantization/stablehlo/cc:report to use TFQuantDialect.

  #92429 merged May 1, 2025

• [xla:cpu:benchmarks] Add a tool to extract dots from HLO text file for CPU backend's dot_benchmark_test

  #92058 merged Apr 30, 2025

• Adds a new function in subgraph to return the pointer directly.

  #92496 merged Apr 30, 2025

219 Pull requests opened by 11 people

• Set `--//third_party/tensorflow/compiler/xla:ci_build=True` for all XLA TAPs.

  #92518 opened Apr 30, 2025

• Automated g4 rollback of changelist 745880173.

  #92519 opened Apr 30, 2025

• Fixes #58106: Add shape validation to AvgPool3DGrad input to prevent crash

  #92521 opened May 1, 2025

• Create CsvDataset.py

  #92522 opened May 1, 2025

• Automated Code Change

  #92527 opened May 1, 2025

• Automated Code Change

  #92528 opened May 1, 2025

• Automated Code Change

  #92529 opened May 1, 2025

• Automated Code Change

  #92530 opened May 1, 2025

• Automated Code Change

  #92531 opened May 1, 2025

• Automated Code Change

  #92534 opened May 1, 2025

• Automated Code Change

  #92535 opened May 1, 2025

• Automated Code Change

  #92536 opened May 1, 2025

• pjrt test fix-1

  #92544 opened May 1, 2025

• pjrt test fix-2

  #92545 opened May 1, 2025

• Automated Code Change

  #92546 opened May 1, 2025

• Automated Code Change

  #92547 opened May 1, 2025

• Rename backend.compile to backend.compile_and_load.

  #92548 opened May 1, 2025

• [XLA:CPU][PJRT] PJRT client handles `run_id` internally.

  #92549 opened May 1, 2025

• #sdy export reduce_scatter collective op in stablehlo export.

  #92551 opened May 1, 2025

• Clarify the semantics of the dimension attribute fields in `LayoutProto`.

  #92553 opened May 1, 2025

• Replace TFL quant dialect with TF quant dialect in register_common_dialects

  #92556 opened May 1, 2025

• Add hermetic `nvshmem` implementation.

  #92563 opened May 1, 2025

• Switch uses of `shape.DebugString()` to `shape.ToString()` or `shape.ToProto().DebugString()`.

  #92572 opened May 1, 2025

• PR #23884: [NVIDIA] Fix a cast error in checking overlap limit function in LHS

  #92573 opened May 1, 2025

• pjrt test fix-2

  #92575 opened May 1, 2025

• Install extra dependencies for pre-release python 3.14 on linux x86 docker images

  #92576 opened May 1, 2025

• Support `np.asarray` for JAX arrays with fully replicated shardings when some hosts have no local shards. Only hosts with local shards can fetch np arrays.

  #92581 opened May 1, 2025

• Automated Code Change

  #92587 opened May 2, 2025

• AllReduce,Case,If,Map,Reduce,Return,While : Direct StableHLO to HLO Translation.

  #92590 opened May 2, 2025

• AllToAll,BatchNormGrad,BatchNormTraining,BitcastConvert,Clamp,CollectiveBroadcast,Compare,Sort : Direct StableHLO to HLO Translation

  #92591 opened May 2, 2025

• Composite,CustomCall,DotGeneral,Dot,DynamicConv,DynamicGather,DynamicIota,DynamicPad,DynamicReshape,Reshape: Direct StableHLO to HLO Translation

  #92592 opened May 2, 2025

• Automated Code Change

  #92593 opened May 2, 2025

• Automated Code Change

  #92594 opened May 2, 2025

• Automated Code Change

  #92595 opened May 2, 2025

• Automated Code Change

  #92596 opened May 2, 2025

• Automated Code Change

  #92597 opened May 2, 2025

• Automated Code Change

  #92598 opened May 2, 2025

• Automated Code Change

  #92600 opened May 2, 2025

• Automated Code Change

  #92602 opened May 2, 2025

• Enable INT32 embeddings in SparseCore

  #92605 opened May 2, 2025

• Automated Code Change

  #92608 opened May 2, 2025

• Automated Code Change

  #92609 opened May 2, 2025

• Automated Code Change

  #92613 opened May 2, 2025

• PR #25953: [XLA:FFI][BugFix] Add Token Args and Rets for typed FFI on CPU backend

  #92615 opened May 2, 2025

• Fix a bug in `RandomPathGreedy` when tracking the objective value over iterations; previously, it adds `cost_delta` to `current_cost` even if a new strategy is rejected due to the peak-memory violation. Also, streamline the previous CL by combining the previous two phases (Phase I and II).

  #92622 opened May 2, 2025

• Remove the lite/toco protobuf files out of //tensorflow/core:lib_internal_impl

  #92624 opened May 2, 2025

• Integrate LLVM at llvm/llvm-project@5060f08c3a98

  #92625 opened May 2, 2025

• A pass interface for verifying that user-guided fusions have formed correctly.

  #92626 opened May 2, 2025

• Reverts 4e2dfdbb4e82695f868a90e823f9a8027ba48a4d

  #92629 opened May 3, 2025

• [XLA] Handle explicit parameter replication during analysis

  #92634 opened May 3, 2025

• Implement Hadamard rotation reference as a custom op

  #92635 opened May 3, 2025

• Add E2E tests for Hadamard's supported ops

  #92636 opened May 3, 2025

• Automated Code Change

  #92637 opened May 3, 2025

• Automated Code Change

  #92638 opened May 3, 2025

• Update get_live_nodes RPC in to make handle worker restarts

  #92639 opened May 3, 2025

• Automated Code Change

  #92640 opened May 3, 2025

• Automated Code Change

  #92641 opened May 3, 2025

• Automated Code Change

  #92642 opened May 3, 2025

• Automated Code Change

  #92643 opened May 3, 2025

• Automated Code Change

  #92644 opened May 3, 2025

• Automated Code Change

  #92646 opened May 3, 2025

• Automated Code Change

  #92649 opened May 3, 2025

• Automated Code Change

  #92651 opened May 3, 2025

• Automated Code Change

  #92652 opened May 3, 2025

• Automated Code Change

  #92653 opened May 3, 2025

• Automated Code Change

  #92654 opened May 3, 2025

• Automated Code Change

  #92655 opened May 3, 2025

• Automated Code Change

  #92656 opened May 3, 2025

• Automated Code Change

  #92657 opened May 3, 2025

• Automated Code Change

  #92658 opened May 3, 2025

• Automated Code Change

  #92662 opened May 4, 2025

• Automated Code Change

  #92663 opened May 4, 2025

• Automated Code Change

  #92664 opened May 4, 2025

• Automated Code Change

  #92665 opened May 4, 2025

• Automated Code Change

  #92666 opened May 4, 2025

• Automated Code Change

  #92667 opened May 4, 2025

• Automated Code Change

  #92668 opened May 4, 2025

• Automated Code Change

  #92669 opened May 4, 2025

• Automated Code Change

  #92670 opened May 4, 2025

• Automated Code Change

  #92671 opened May 4, 2025

• Automated Code Change

  #92672 opened May 4, 2025

• Automated Code Change

  #92673 opened May 4, 2025

• Automated Code Change

  #92674 opened May 4, 2025

• Automated Code Change

  #92675 opened May 4, 2025

• Automated Code Change

  #92677 opened May 4, 2025

• Automated Code Change

  #92678 opened May 4, 2025

• Automated Code Change

  #92679 opened May 4, 2025

• Automated Code Change

  #92680 opened May 4, 2025

• Automated Code Change

  #92681 opened May 4, 2025

• Automated Code Change

  #92682 opened May 4, 2025

• Automated Code Change

  #92683 opened May 4, 2025

• compat: Update forward compatibility horizon to 2025-05-04

  #92684 opened May 4, 2025

• Get rid of `HloModuleProto` from `CompiledMemoryStats`

  #92690 opened May 4, 2025

• Automated Code Change

  #92692 opened May 5, 2025

• Automated Code Change

  #92693 opened May 5, 2025

• Automated Code Change

  #92694 opened May 5, 2025

• Automated Code Change

  #92695 opened May 5, 2025

• Automated Code Change

  #92696 opened May 5, 2025

• Automated Code Change

  #92697 opened May 5, 2025

• Automated Code Change

  #92698 opened May 5, 2025

• Automated Code Change

  #92699 opened May 5, 2025

• Automated Code Change

  #92700 opened May 5, 2025

• Automated Code Change

  #92701 opened May 5, 2025

• TfLite elementwise_ops add type support (#104)

  #92706 opened May 5, 2025

• TfLite exp missing datatype support (#69) (#121)

  #92714 opened May 5, 2025

• Explicitly include `errors.h` in `status_macros.h`. We need `tsl::error::INTERNAL` from it.

  #92723 opened May 5, 2025

• bug fix in literal comparison in `equal` test.

  #92724 opened May 5, 2025

• Check whether createStream was successful

  #92729 opened May 5, 2025

• remove transitive lite dependencies from tf_passes

  #92731 opened May 5, 2025

• [xla::PyClient] Update PyClient to use xla::ifrt::CompileAndLoad.

  #92732 opened May 5, 2025

• PR #25635: Add parsing debug options support

  #92734 opened May 5, 2025

• [XLA:GPU] Return an error if `GetSchedulerMemoryLimit` fails.

  #92735 opened May 5, 2025

• [Cleanup] Cleanup whitespace

  #92741 opened May 5, 2025

• Collect execution time for mlrt.

  #92744 opened May 5, 2025

• [Cleanup] Use CHECK_NOTNULL and CHECK_OK

  #92745 opened May 5, 2025

• Load python rules instead of using native rules.

  #92746 opened May 5, 2025

• Get rid of `HloModuleProto` from `CompiledMemoryStats`

  #92748 opened May 5, 2025

• Fix clang-format and BUILD dependency issues

  #92751 opened May 5, 2025

• Update GetLiveNodes API to be a bit more robust to worker restarts.

  #92752 opened May 5, 2025

• Integrate LLVM at llvm/llvm-project@0009a1783490

  #92754 opened May 5, 2025

• Automated Code Change

  #92756 opened May 5, 2025

• [JAX] Make fully replicated sharding to avoid materializing the same host buffers

  #92759 opened May 6, 2025

• Automated Code Change

  #92760 opened May 6, 2025

• Automated Code Change

  #92761 opened May 6, 2025

• Automated Code Change

  #92762 opened May 6, 2025

• Automated Code Change

  #92764 opened May 6, 2025

• Automated Code Change

  #92765 opened May 6, 2025

• Fix missing dep.

  #92766 opened May 6, 2025

• Automated Code Change

  #92767 opened May 6, 2025

• Delete constructor for tf_allocator_adapter.

  #92768 opened May 6, 2025

• Fixed ppc64le build issues

  #92769 opened May 6, 2025

• Update registration.h

  #92771 opened May 6, 2025

• Automated Code Change

  #92773 opened May 6, 2025

• Set wheel CUDA versions to be minor-version lower bound.

  #92774 opened May 6, 2025

• Automated Code Change

  #92775 opened May 6, 2025

• [XLA:GPU][Emitters] Use new heuristic for transpose emitter

  #92776 opened May 6, 2025

• Use transpose emitter for transposes with 4-bits datatypes

  #92777 opened May 6, 2025

• PR #26100: Clean up special cases for AsyncStart in HloComputation.

  #92778 opened May 6, 2025

• Added min/max reductions on XNNPACK delegate level.

  #92782 opened May 6, 2025

• [mlir][tosa] Fix build error

  #92788 opened May 6, 2025

• Add basic thunk deserialization logic

  #92793 opened May 6, 2025

• [xla:gpu] Fork existing fusion_emitter_int4_device_test.cc and make it use generic emitter.

  #92801 opened May 6, 2025

• Add [de]serialization for `GemmConfig`.

  #92802 opened May 6, 2025

• Set remaining profile options and perform validation on the advanced_configuration keys.

  #92804 opened May 6, 2025

• Add [de]serialization for `BufferAllocationSlice`.

  #92805 opened May 6, 2025

• Add proto serialization for GPU CopyThunk

  #92808 opened May 6, 2025

• [XLA:Collective] Cache the results from GetParticipatingFlattenedIdGroups().

  #92813 opened May 6, 2025

• Add specialization for `PrintToStream` in `TF_RET_CHECK`'s additional output.

  #92814 opened May 6, 2025

• Reverts 6932184196b02be51a5be4ca1090a422f897dc4e

  #92817 opened May 6, 2025

• Integrate LLVM at llvm/llvm-project@3e235a7c601d

  #92818 opened May 6, 2025

• Make Shape reference-counted with copy-on-write.

  #92819 opened May 6, 2025

• Replaced the GreedySubGraphExactMatcher with a new GreedyTopDownMatcher that employs stricter matching criteria. This new matcher only propagates mappings between nodes that have the same number of children, and whose corresponding children (in the same position) share the same type.

  #92820 opened May 6, 2025

• Add benchmark for Shape assignment.

  #92824 opened May 6, 2025

• Add an API to compilation environments to initialize all known compilation environments

  #92825 opened May 6, 2025

• Replace `std::shared_ptr<xla::ifrt::LoadedExecutable>` with `xla::ifrt::LoadedExecutableRef`

  #92827 opened May 7, 2025

• Prevent MSA from creating a worst case unbalanced buffer interval tree when allocating scoped vmem.

  #92829 opened May 7, 2025

• Adds checkpoint test for tpu_embedding_v3

  #92831 opened May 7, 2025

• Use utility functions to check number of inputs and outputs.

  #92832 opened May 7, 2025

• Automated Code Change

  #92833 opened May 7, 2025

• Automated Code Change

  #92834 opened May 7, 2025

• Automated Code Change

  #92835 opened May 7, 2025

• Automated Code Change

  #92836 opened May 7, 2025

• Automated Code Change

  #92837 opened May 7, 2025

• Automated Code Change

  #92838 opened May 7, 2025

• Automated Code Change

  #92839 opened May 7, 2025

• Automated Code Change

  #92840 opened May 7, 2025

• Automated Code Change

  #92841 opened May 7, 2025

• Automated Code Change

  #92842 opened May 7, 2025

• Automated Code Change

  #92843 opened May 7, 2025

• Automated Code Change

  #92844 opened May 7, 2025

• Automated Code Change

  #92845 opened May 7, 2025

• Automated Code Change

  #92846 opened May 7, 2025

• Automated Code Change

  #92847 opened May 7, 2025

• Automated Code Change

  #92848 opened May 7, 2025

• Automated Code Change

  #92849 opened May 7, 2025

• Automated Code Change

  #92850 opened May 7, 2025

• Automated Code Change

  #92851 opened May 7, 2025

• Automated Code Change

  #92852 opened May 7, 2025

• Automated Code Change

  #92853 opened May 7, 2025

• Automated Code Change

  #92854 opened May 7, 2025

• Automated Code Change

  #92855 opened May 7, 2025

• [XLA:GPU] Add simple triton support test for topK op

  #92856 opened May 7, 2025

• Automated Code Change

  #92858 opened May 7, 2025

• Automated Code Change

  #92859 opened May 7, 2025

• Automated Code Change

  #92861 opened May 7, 2025

• Automated Code Change

  #92864 opened May 7, 2025

• Automated Code Change

  #92865 opened May 7, 2025

• [xla:gpu] add extract_tma_info pass

  #92867 opened May 7, 2025

• Update TmaDescriptor creation to use triton's internal swizzling decision

  #92868 opened May 7, 2025

• Automated Code Change

  #92870 opened May 7, 2025

• Automated Code Change

  #92871 opened May 7, 2025

• Reverts f6269e13f1d709b4fe46079e36f1407f65075271

  #92874 opened May 7, 2025

• Reenable tma_to_llvm.mlir test

  #92875 opened May 7, 2025

• Fix: Prevent Heap-Buffer-Overflow in stringprintf_fuzz Fuzzer

  #92877 opened May 7, 2025

• feat: add datatype support for add, ceil, mul, range, sign, sub

  #92880 opened May 7, 2025

• PR #26183: [XLA:GPU] Add support for SM103a and SM121a architectures (Blackwell)

  #92881 opened May 7, 2025

• [XLA:GPU] Use cutlass 3.8.0

  #92884 opened May 7, 2025

• #sdy Cleanup `removeAutoAxesToAvoidPadding` as this is now done in `-sdy-update-non-divisible-input-output-shardings`.

  #92885 opened May 7, 2025

• [XLA:CPU] Set fp-contract to on rather than fast for DUCC

  #92887 opened May 7, 2025

• Re-enable test that is now passing.

  #92888 opened May 7, 2025

• [xla:ifrt] Rename references to ::Compile(module, topology, options) that return LoadedExecutables to ::CompileAndLoad.

  #92892 opened May 7, 2025

• [XLA:CPU][roll-forward] Internal fixes.

  #92894 opened May 7, 2025

• fork quantize_passes to use tf_passes

  #92895 opened May 7, 2025

• Integrate LLVM at llvm/llvm-project@ed8d4fe0b1d8

  #92896 opened May 7, 2025

• Add To/FromProto() to BufferAllocation::Slice

  #92897 opened May 7, 2025

• Rename tf_quant ops similar to original quant ops

  #92899 opened May 7, 2025

• Custom NVPTX backend.

  #92900 opened May 7, 2025

• Integrate LLVM at llvm/llvm-project@2d287f51eff2

  #92901 opened May 7, 2025

• Introduce the `BUFFER` primitive type and update the `Shape` class to support it.

  #92906 opened May 7, 2025

• Integrate StableHLO at openxla/stablehlo@630c315b

  #92907 opened May 7, 2025

• Fix TFLite Compiler warnings in Chromium build

  #92908 opened May 7, 2025

• Branch quantization/stablehlo/cc:pre_calibration to use TFQuantDialect

  #92909 opened May 7, 2025

• Add boolean attribute use_shardy_partitioner to XlaCallModule op.

  #92910 opened May 7, 2025

• [mlir][tosa] Update variable read/write names to align with v1.0 spec

  #92911 opened May 7, 2025

• [XLA:benchmarks] Add workflow files to generate GHA input matrices and a presubmit workflow

  #92912 opened May 7, 2025

• Provide flexibility to derived classes for specific backends w.r.t how the initial fusion worklist is formed for a current computation.

  #92913 opened May 7, 2025

• Remove support for sparse dimension's ordered attribute.

  #92914 opened May 7, 2025

• Deprecate `*_size()` methods to reduce the `Layout` API surface.

  #92915 opened May 7, 2025

• Add traceme while borrowing streams

  #92916 opened May 7, 2025

13 Issues closed by 3 people

• Install TensorFlow with pip for macOS doesn't work on M4 MacBook Pro Apple Silicon

  #91681 closed May 7, 2025

• Tensorflow model.save fails with AttributeError: 'Variable' object has no attribute 'read_value'

  #91686 closed May 7, 2025

• Loading tensorflow DLL error in Spyder other python ok

  #91709 closed May 7, 2025

• Mac Air M2, TfLiteGpuDelegate, Cpp, C++ Building from source

  #86075 closed May 6, 2025

• CMake Error: install(EXPORT "tensorflow-liteTargets" ...) includes target "tensorflow-lite" which requires target "pthreadpool" that is not in any export set.

  #87172 closed May 6, 2025

• FP32 gives good results but int8 predicts only single class

  #89438 closed May 6, 2025

• Tensorflow Build Error

  #91558 closed May 6, 2025

• multiprocessing incompatibility

  #91647 closed May 6, 2025

• MemoryError during "import tensorflow" with tensorflow-cpu==2.19.0

  #89268 closed May 5, 2025

• Tensor flow

  #92685 closed May 4, 2025

• Failed to load the native TensorFlow runtime.

  #92650 closed May 3, 2025

• Linking fails with "Undefined reference to riegeli::RecordsMetadata::Clear()"

  #91314 closed May 2, 2025

• Efficientnet - TFLite Inference Issue

  #89556 closed May 1, 2025

10 Issues opened by 8 people

• `tf.math.imag` doesn't support float and double

  #92876 opened May 7, 2025

• `tf.math.igamma` doesn't work when dtype is bfloat16 or half

  #92873 opened May 7, 2025

• Compatibility Challenges of Transformer Models (MobileBERT, Mobile ViT) with TensorFlow Lite and GPU Delegation

  #92798 opened May 6, 2025

• Different inference results on CPU vs GPU for simple model on macOS M1

  #92686 opened May 4, 2025

• Problem when using tf-to-tosa-pipeline

  #92660 opened May 3, 2025

• Error when converting tensorflow model in mlir

  #92659 opened May 3, 2025

• Setting the protobuf version at installation with bazel

  #92610 opened May 2, 2025

• TensorFlow 2.12, 2.16.2,2.19 using M4 Mac vs colab GPU give incorrect accuracy and >2.12TF are slower

  #92564 opened May 1, 2025

• Tensorflow for Arch Linux

  #92550 opened May 1, 2025

• No way to use tensorflow with cuda on windows. cudaGetDevice() failed.

  #92523 opened May 1, 2025

61 Unresolved conversations

Sometimes conversations happen on old items that aren’t yet closed. Here is a list of all the Issues and Pull Requests with unresolved conversations.

• Tensorflow import failed

  #91656 commented on Apr 30, 2025 • 0 new comments

• Feature Request: GPUOptions for Go binding

  #22926 commented on May 1, 2025 • 0 new comments

• Compiling TFLite introductory example fails due to missing headers

  #91782 commented on May 1, 2025 • 0 new comments

• warning in official

  #91733 commented on May 1, 2025 • 0 new comments

• It doesn't support on python3.13

  #78774 commented on May 1, 2025 • 0 new comments

• "Given shapes (...) are not broadcastable" after conversion onnx->tf->tflite

  #92037 commented on May 2, 2025 • 0 new comments

• TensorFlow Lite Model Maker link is broken in this codelab

  #92191 commented on May 2, 2025 • 0 new comments

• Compiling 2.19.0 through cmake on Windows, there are too many exported DLL symbols and an increase in disk size

  #92275 commented on May 2, 2025 • 0 new comments

• Failed to create Hexagon delegate

  #92488 commented on May 2, 2025 • 0 new comments

• The pip instructions to download and install tensorflow for macOS don't work

  #92062 commented on May 3, 2025 • 0 new comments

• Unable to install TensorFlow: No matching distribution found for TensorFlow!

  #79349 commented on May 3, 2025 • 0 new comments

• GPU kernel for `tf.linalg.eig` / `tf.linalg.eigvals`

  #92194 commented on May 5, 2025 • 0 new comments

• New version of nVidia Windows GPU driver (576+) causing TF in WSL process to crash.

  #92251 commented on May 5, 2025 • 0 new comments

• TensorFlow on RTX 5090

  #89272 commented on May 5, 2025 • 0 new comments

• The default value for skip_gradients_aggregation (False) of optimizers prevents freeing memory

  #92094 commented on May 5, 2025 • 0 new comments

• Impossible to free GPU memory used by rank 0 tensors

  #92103 commented on May 5, 2025 • 0 new comments

• How to setup tensorflow environment in google colab for training ai models using tensorflow object detection APIs?

  #91738 commented on May 6, 2025 • 0 new comments

• [TFLite] GPU delegation issues with Vision Transformer (ViT-base) in TFLite: dynamic shape & op support problems

  #92098 commented on May 6, 2025 • 0 new comments

• Muting Tensorflow Lite logs

  #92216 commented on May 6, 2025 • 0 new comments

• model load problem

  #92044 commented on May 7, 2025 • 0 new comments

• Compiling Tensorflow v2.19.0 with mkl_aarch64 fails.

  #92279 commented on May 7, 2025 • 0 new comments

• GPU Illegal memory access in kernel DenseToCSRSparseMatrix

  #92413 commented on May 7, 2025 • 0 new comments

• Add ComplexOp as a builtin to TFLite

  #61359 commented on May 5, 2025 • 0 new comments

• Support for arbitrary tensor sizes in `tfl.strided_slice` lowering

  #61424 commented on May 5, 2025 • 0 new comments

• Build kernel for all numbertypes for ApproximateEqual Op

  #64587 commented on May 5, 2025 • 0 new comments

• Fix checkfail in MatrixSetDiagV3

  #65702 commented on May 5, 2025 • 0 new comments

• iwyu: missing includes in xla

  #66450 commented on May 5, 2025 • 0 new comments

• Add GPU configuration options to TensorFlow Go bindings

  #81989 commented on May 7, 2025 • 0 new comments

• Remove Transfer-Encoding: identity considered to be illegal

  #83082 commented on May 5, 2025 • 0 new comments

• [oneDNN][CPU] fuse a matmul pattern

  #86172 commented on May 5, 2025 • 0 new comments

• fix(kernels): Handle empty values with non-empty row splits in RaggedTensorToTensor

  #86349 commented on May 5, 2025 • 0 new comments

• Fix compile error in tensorflow/python/tfcompile_wrapper.cc on s390x

  #87676 commented on May 5, 2025 • 0 new comments

• TfLite. Fix of issue 61269

  #87946 commented on May 5, 2025 • 0 new comments

• TfLite. Fix of issue 54269

  #88020 commented on May 5, 2025 • 0 new comments

• [XLA:CPU][tfcompile] Use thunk runtime for tfcompiled models.

  #88552 commented on May 6, 2025 • 0 new comments

• Fix: Ensure boolean_mask_v2() only accepts boolean dtype for mask

  #89370 commented on May 5, 2025 • 0 new comments

• Move duplicate CUDA/XLA registration logs from INFO to VLOG

  #89808 commented on May 5, 2025 • 0 new comments

• [XLA:GPU] Remove CollectivePermuteValidIterationAnnotator pass

  #89923 commented on May 6, 2025 • 0 new comments

• [XLA:GPU] Deprecate PIPELINE_PARALLELISM_OPT_LEVEL_ENABLE_CYCLE_DECOMPOSER

  #90010 commented on May 6, 2025 • 0 new comments

• Internal dir restructure

  #90497 commented on May 7, 2025 • 0 new comments

• Adapt commit 5467ee9 from XLA to TensorFlow

  #90563 commented on May 6, 2025 • 0 new comments

• Create a PjRt protos dir.

  #90719 commented on May 7, 2025 • 0 new comments

• adam opt .

  #90795 commented on May 5, 2025 • 0 new comments

• #sdy Properly handle token types in JAX and `ManualComputationOp`.

  #91048 commented on May 7, 2025 • 0 new comments

• Enable Android XNNPACK delegate num_threads with Interpreter.Options.…

  #91181 commented on May 6, 2025 • 0 new comments

• Cleaned up coordination service key-value store.

  #91424 commented on May 1, 2025 • 0 new comments

• Implement psend and precv in lax/parallel.py

  #91682 commented on May 1, 2025 • 0 new comments

• experimental test for dup registration bug

  #91758 commented on May 7, 2025 • 0 new comments

• Cleanup unused visibility rules for XProf

  #92014 commented on May 2, 2025 • 0 new comments

• Don't track dimension attributes separately in Layout.

  #92096 commented on May 1, 2025 • 0 new comments

• Branch tf_test_passes to use TFQuantDialect

  #92138 commented on May 7, 2025 • 0 new comments

• Optimize `xla::HloSharding::PartialTile` by using vectorized highway sort `hwy::VQSort`.

  #92165 commented on May 3, 2025 • 0 new comments

• Major deps udpate:

  #92241 commented on May 7, 2025 • 0 new comments

• Support ZeroCorp custom_op as a formatting step in hlo sharding utils.

  #92315 commented on May 1, 2025 • 0 new comments

• [XLA:ReplicaGroupV2] Propagate ReplicaGroup to more places. (NFC)

  #92319 commented on May 7, 2025 • 0 new comments

• Remove unnecessary dependency.

  #92337 commented on May 2, 2025 • 0 new comments

• [XLA:GPU][Emitters] Scan through the IR to compute shared memory usage.

  #92427 commented on May 5, 2025 • 0 new comments

• PR #25707: [XLA:GPU] Fix cuDNN command buffer incorrect update

  #92468 commented on May 1, 2025 • 0 new comments

• feat: add datatype support for tfl.transpose

  #92472 commented on May 6, 2025 • 0 new comments

• Do not create BFCAllocator if not needed

  #92483 commented on May 5, 2025 • 0 new comments

• Update Riegeli dependency to the latest commit.

  #92502 commented on May 7, 2025 • 0 new comments