openxla/xla issues and pull requests

#22658 - Plumb layout through the creation of PjRtArrays.

Pull Request - State: open - Opened by copybara-service[bot] 5 days ago

#22657 - [NCCL] Fix ras build issue in OSS.

Pull Request - State: closed - Opened by copybara-service[bot] 5 days ago

#22656 - Add stale check for debugging segfault.

Pull Request - State: open - Opened by copybara-service[bot] 5 days ago

#22655 - Move ARM CI back to `c4a`

Pull Request - State: open - Opened by copybara-service[bot] 5 days ago

#22654 - [IFRT] Add log message to CHECK in compile_atom_program_pass to aid with debugging

Pull Request - State: closed - Opened by copybara-service[bot] 5 days ago

#22653 - Make simpler to reason about expansion logic from Iota to legacy replica groups.

Pull Request - State: open - Opened by copybara-service[bot] 5 days ago

#22652 - Do not add control dependency from send to recv-done in decomposed collective-permute

Pull Request - State: open - Opened by copybara-service[bot] 5 days ago

#22651 - Merge pjrt_stream_executor_client.h and gpu/se_gpu_pjrt_client.h into

Pull Request - State: open - Opened by copybara-service[bot] 5 days ago

#22650 - Add AllReduceInfo to step data for TPUs.

Pull Request - State: open - Opened by copybara-service[bot] 5 days ago

#22649 - Integrate LLVM at llvm/llvm-project@165a3d6a9b16

Pull Request - State: open - Opened by copybara-service[bot] 5 days ago

#22648 - Internal visibility changes

Pull Request - State: closed - Opened by copybara-service[bot] 5 days ago

#22647 - Use IsOkAndHolds in all-reduce combiner test

Pull Request - State: open - Opened by copybara-service[bot] 5 days ago

#22646 - Documents when to use xla_test/xla_cc_test/tsl_cc_test.

Pull Request - State: closed - Opened by copybara-service[bot] 5 days ago

#22645 - Fix error in the gpu_specs README. The spec is the TargetConfig, which includes the device description.

Pull Request - State: open - Opened by dimvar 5 days ago

#22645 - Fix error in the gpu_specs README. The spec is the TargetConfig, which includes the device description.

Pull Request - State: open - Opened by dimvar 5 days ago

#22644 - Modify the hash value for CseKey to also hash on result accuracy. The cse pass should also group instructions with same result accuracy. We should also check for result accuracy fields when checking if two hlo instructions are the same.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22644 - Modify the hash value for CseKey to also hash on result accuracy. The cse pass should also group instructions with same result accuracy. We should also check for result accuracy fields when checking if two hlo instructions are the same.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22643 - Add a REPO.bazel file to indicate this is a Bazel repo

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22643 - Add a REPO.bazel file to indicate this is a Bazel repo

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22642 - Do not combine all-reduces with control dependencies.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22642 - Do not combine all-reduces with control dependencies.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22641 - Replace macOS x86 presubmits with macOS Arm64 presubmits.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22641 - Replace macOS x86 presubmits with macOS Arm64 presubmits.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22640 - [NFC] Deduplicate functions between HLO runners.

Pull Request - State: open - Opened by sergachev 6 days ago

#22640 - [NFC] Deduplicate functions between HLO runners.

Pull Request - State: open - Opened by sergachev 6 days ago

#22639 - Only pipeline send/recv if this does not violate control dependencies

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22639 - Only pipeline send/recv if this does not violate control dependencies

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22638 - Reenable pipeline parallelism test after fix

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22638 - Reenable pipeline parallelism test after fix

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22637 - Support creating HloModule from proto with compilation environments.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22637 - Support creating HloModule from proto with compilation environments.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22636 - [XLA:GPU][TRITON:XLA] Add ops and types to support TMA.

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22636 - [XLA:GPU][TRITON:XLA] Add ops and types to support TMA.

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22635 - Upgrade to Bazel 7.x

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22635 - Upgrade to Bazel 7.x

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22634 - PR #22541: [ROCm] Cleanup atomics support

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22634 - PR #22541: [ROCm] Cleanup atomics support

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22633 - Reverts 22f9c10e9c2781485f5d12e6525258414dd3a67f

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22632 - [pjrt] Removed PjRtDevice overloads of `PjRtClient::CreateBuffersForAsyncHostToDevice`

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22632 - [pjrt] Removed PjRtDevice overloads of `PjRtClient::CreateBuffersForAsyncHostToDevice`

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22631 - [XLA:GPU] Rewrite if else chains into switch statements.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22631 - [XLA:GPU] Rewrite if else chains into switch statements.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22630 - Integrate LLVM at llvm/llvm-project@0e779ad4998e

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22630 - Integrate LLVM at llvm/llvm-project@0e779ad4998e

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22629 - #sdy Add JAX backwards compatibility test.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22629 - #sdy Add JAX backwards compatibility test.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22628 - Enable test and make it pass.

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22628 - Enable test and make it pass.

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22627 - Change `DeviceList::FromProto()` to use `Client::MakeDeviceList()` to create device lists

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22626 - Automated Code Change

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22626 - Automated Code Change

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22625 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22625 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22624 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22624 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22623 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22623 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22622 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22622 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22621 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22621 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22620 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22620 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22619 - PR #22532: [ROCm] Use device allocate for collective memory allocation

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22619 - PR #22532: [ROCm] Use device allocate for collective memory allocation

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22618 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22618 - Automated Code Change

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22617 - PR #22512: [XLA:GPU] Enable cuDNN kernel for NVFP4 block scaled dot

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22617 - PR #22512: [XLA:GPU] Enable cuDNN kernel for NVFP4 block scaled dot

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22616 - Integrate LLVM at llvm/llvm-project@ab93bd6959d6

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22616 - Integrate LLVM at llvm/llvm-project@ab93bd6959d6

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22615 - [XLA:GPU] Add a debug option `xla_gpu_unsupported_force_triton_gemm` for use

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22615 - [XLA:GPU] Add a debug option `xla_gpu_unsupported_force_triton_gemm` for use

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22614 - Fix hlo_opt printing of Hlo module

Pull Request - State: open - Opened by shraiysh 6 days ago

#22613 - Replace the remaining use of `BasicDeviceList` in the base IFRT

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22612 - [XLA]Further increase timeout to 6hr for CPU benchmarks

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22612 - [XLA]Further increase timeout to 6hr for CPU benchmarks

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22610 - HloModule is converted to proto in MlirToXlaComputation so need to make sure result accuracy is kept.

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22607 - [XLA:GPU] Make command_buffer_thunk_test:DynamicSliceFusionCmd more tolerant

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22605 - [XLA] Add tests for hlo_pass_fix.

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22601 - Support optimization_level and memory_fitting_level XLA compilation options.

Pull Request - State: open - Opened by copybara-service[bot] 6 days ago

#22593 - [XLA:GPU] Fix triton dot op on sm120 (RTX50xx)

Pull Request - State: open - Opened by sergey-kozub 6 days ago

#22593 - [XLA:GPU] Fix triton dot op on sm120 (RTX50xx)

Pull Request - State: open - Opened by sergey-kozub 6 days ago

#22589 - [XLA:GPU] Move ConvertRangeVariablesToDimensions from scatter.cc to indexing_map.h.

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22589 - [XLA:GPU] Move ConvertRangeVariablesToDimensions from scatter.cc to indexing_map.h.

Pull Request - State: closed - Opened by copybara-service[bot] 6 days ago

#22586 - Only cache jax.Array._npy_value when a copy is required.

Pull Request - State: open - Opened by copybara-service[bot] 7 days ago

#22586 - Only cache jax.Array._npy_value when a copy is required.

Pull Request - State: open - Opened by copybara-service[bot] 7 days ago

#22584 - PR #22554: [GPU] Fix cuDNN fusion compiler support of control predecessors.

Pull Request - State: open - Opened by copybara-service[bot] 7 days ago

#22584 - PR #22554: [GPU] Fix cuDNN fusion compiler support of control predecessors.

Pull Request - State: open - Opened by copybara-service[bot] 7 days ago

#22583 - [xla:cpu:onednn] Enable oneDNN thread pool targets

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22583 - [xla:cpu:onednn] Enable oneDNN thread pool targets

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22582 - Prepare support for multiple fusion roots.

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22582 - Prepare support for multiple fusion roots.

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22581 - PR #22529: [ROCm] Fix link error with rccl by adding libnuma link dependency

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22581 - PR #22529: [ROCm] Fix link error with rccl by adding libnuma link dependency

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22580 - PR #22537: [GPU] Add cuDNN 9.7.1.

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22580 - PR #22537: [GPU] Add cuDNN 9.7.1.

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22579 - PR #22509: [ROCm] Avoid hardcoding hipcc compiler includes

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22579 - PR #22509: [ROCm] Avoid hardcoding hipcc compiler includes

Pull Request - State: closed - Opened by copybara-service[bot] 7 days ago

#22576 - PR #20332: [ds-fusion] Add runtime support for host calculation of offsets in ds fusion

Pull Request - State: open - Opened by copybara-service[bot] 7 days ago

GitHub / openxla/xla issues and pull requests