openxla/xla issues and pull requests

#22798 - Add a field in PluginCompileOptions to specify the input program format.

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

#22797 - Automated Code Change

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

#22797 - Automated Code Change

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

#22793 - Create GitHub action to run a small GPU HLO benchmark and save XSpace data.

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

#22787 - PR #22782: [NFC] Move HLO-handling functions from runner to module util.

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

#22774 - [pjrt] Removed `PjRtStreamExecutorLoadedExecutable::IsReturnedFutureSupported`

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

#22773 - [pjrt] Removed unused `PjRtBuffer::CopyToRemoteDeviceScattered` and `ScatterDetails`

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

#22773 - [pjrt] Removed unused `PjRtBuffer::CopyToRemoteDeviceScattered` and `ScatterDetails`

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

#22766 - [pjrt] Removed unused `PjRtClient::MakeCrossHostReceiveBufferForGather` and `GatherDetails`

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

#22766 - [pjrt] Removed unused `PjRtClient::MakeCrossHostReceiveBufferForGather` and `GatherDetails`

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

#22759 - Tweak formatting of a few lines in `generate_hlo_test_checks.py`.

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

#22759 - Tweak formatting of a few lines in `generate_hlo_test_checks.py`.

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

#22754 - Call out `generate_hlo_test_checks.py` in online documentation.

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

#22754 - Call out `generate_hlo_test_checks.py` in online documentation.

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

#22753 - [XLA:MSA] Fix an issue in MSA where `operands_in_alternate_memory_map_` was not correctly updated when there is cross program prefetch.

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

#22750 - [StableHLO] Only emit StableHLO from xla_computation_to_mlir_module

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

#22750 - [StableHLO] Only emit StableHLO from xla_computation_to_mlir_module

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

#22748 - [RFC] [XLA][TPU][MSA] Use HW Spec instead of hardcoding bandwidth multipliers

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

#22747 - [XLA] Fix a bug in SchedulingAnnotationCrossesOverlapLimit.

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

#22747 - [XLA] Fix a bug in SchedulingAnnotationCrossesOverlapLimit.

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

#22746 - Move utility aspect rules to the `third_party/py/python_wheel.bzl`.

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

#22746 - Move utility aspect rules to the `third_party/py/python_wheel.bzl`.

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

#22745 - Create JAX wheel build target.

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

#22745 - Create JAX wheel build target.

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

#22717 - Fix kernel_thunk_test compilation error

Pull Request - State: closed - Opened by apivovarov 7 days ago - 4 comments

#22717 - Fix kernel_thunk_test compilation error

Pull Request - State: closed - Opened by apivovarov 7 days ago - 4 comments

#22714 - Add conversion from quantization composites to stablehlo native quantized ops.

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

#22712 - Migrate profiler code from tensorflow repo to profiler repo

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

#22712 - Migrate profiler code from tensorflow repo to profiler repo

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

#22696 - HloRunnerPjRt should respect static device layout if present.

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

#22696 - HloRunnerPjRt should respect static device layout if present.

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

#22675 - [PJRT:CPU] Derive ExecutableRunOptions::run_id from ExecuteOptions::launch_id

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

#22671 - Make xla_test, etc default to linkstatic to catch duplicate symbols at build time.

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

#22658 - Plumb layout through the creation of PjRtArrays.

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

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

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

#22656 - Add stale check for debugging segfault.

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

#22655 - Move ARM CI back to `c4a`

Pull Request - State: open - Opened by copybara-service[bot] 8 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] 8 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] 8 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] 8 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] 8 days ago

#22650 - Add AllReduceInfo to step data for TPUs.

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

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

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

#22648 - Internal visibility changes

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

#22647 - Use IsOkAndHolds in all-reduce combiner test

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

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

Pull Request - State: closed - Opened by copybara-service[bot] 8 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 8 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 8 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] 8 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] 8 days ago

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

#22638 - Reenable pipeline parallelism test after fix

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

#22638 - Reenable pipeline parallelism test after fix

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

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

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

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

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

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

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

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

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

#22635 - Upgrade to Bazel 7.4.1

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

#22635 - Upgrade to Bazel 7.4.1

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

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

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

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

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

#22633 - Reverts 22f9c10e9c2781485f5d12e6525258414dd3a67f

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

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

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

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

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

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

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

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

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

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

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

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

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

#22629 - #sdy Add JAX backwards compatibility test.

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

#22629 - #sdy Add JAX backwards compatibility test.

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

#22628 - Enable test and make it pass.

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

#22628 - Enable test and make it pass.

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

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

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

#22626 - Automated Code Change

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

#22626 - Automated Code Change

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

#22625 - Automated Code Change

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

#22625 - Automated Code Change

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

#22624 - Automated Code Change

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

#22624 - Automated Code Change

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

#22623 - Automated Code Change

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

#22623 - Automated Code Change

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

#22622 - Automated Code Change

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

#22622 - Automated Code Change

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

#22621 - Automated Code Change

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

#22621 - Automated Code Change

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

#22620 - Automated Code Change

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

#22620 - Automated Code Change

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

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

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

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

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

#22618 - Automated Code Change

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

#22618 - Automated Code Change

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

GitHub / openxla/xla issues and pull requests