Skip to content

Pull requests: openxla/xla

New pull request New
2,360 Open 17,734 Closed
2,360 Open 17,734 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[xla:cpu] Add an xnn_threadpool for wrapping ParallelLoopRunner as pthreadpool API
#20542 opened Dec 14, 2024 by copybara-service bot Loading…
@ezhulenev
[xla:cpu] Implement 2D and 3D loop parallelization
#20541 opened Dec 14, 2024 by copybara-service bot Loading…
@ezhulenev
Optimize XLA SPMD Slice partitioner, containing the following 3 steps.
#20540 opened Dec 14, 2024 by copybara-service bot Loading…
@ZixuanJiang
[xla:gpu] Add an option to use persistent collective cliques
#20539 opened Dec 14, 2024 by copybara-service bot Loading…
@ezhulenev
Remove legacy trivial main programs
#20537 opened Dec 13, 2024 by copybara-service bot Loading…
@toli-y
[Shardy] HLO ⇄ MHLO to HLO ⇄ StableHLO
#20536 opened Dec 13, 2024 by copybara-service bot Loading…
@abhigunj
Reverts a8a0a7ec72ed3960b33a061c12e43d7e5beb9087
#20535 opened Dec 13, 2024 by copybara-service bot Loading…
Remove CUDA dependencies from jaxlib wheel.
#20534 opened Dec 13, 2024 by copybara-service bot Loading…
[XLA] Add S1 and U1 as data types
#20533 opened Dec 13, 2024 by copybara-service bot Loading…
@amitsabne1
Make collective selet folder convert-aware
#20532 opened Dec 13, 2024 by copybara-service bot Loading…
@frgossen
Integrate LLVM at llvm/llvm-project@5f72f2c8fd6c
#20531 opened Dec 13, 2024 by copybara-service bot Loading…
Use absl::string_view instead of std::string_view as some environments (e.g. Android) don't provide std::string_view.
#20530 opened Dec 13, 2024 by copybara-service bot Loading…
@klucke
[XLA:GPU] update microbenchmarks to include nccl groups in a while loop
#20528 opened Dec 13, 2024 by copybara-service bot Loading…
[XLA:GPU] update collective pipeline parallelism execution test to include nccl groups in a while loop
#20527 opened Dec 13, 2024 by copybara-service bot Loading…
Temporary check to avoid cases where wrong result sharding causes a RET_CHECK to fail in hlo_sharding.cc.
#20526 opened Dec 13, 2024 by copybara-service bot Loading…
@tomnatan30
Remove CHECKs for inline vectors. Returning 0 is functionally correct.
#20525 opened Dec 13, 2024 by copybara-service bot Loading…
@toli-y
Move some ShapeUtil validation to cpp. Create generic ShapeError.
#20524 opened Dec 13, 2024 by copybara-service bot Loading…
@toli-y
[XLA:GPU] Synchronize compute and communication streams in NCCL group implementation
#20523 opened Dec 13, 2024 by copybara-service bot Loading…
#sdy add constraints to maximal shardings during verification.
#20519 opened Dec 13, 2024 by copybara-service bot Loading…
@bartchr808
#sdy make mhlo_export_pipeline test use no rank maximal sharding.
#20518 opened Dec 13, 2024 by copybara-service bot Loading…
@bartchr808
Simplify WaitAndLogIfStuck() to use absl::Barrier instead of tsl::BlockingCounter.
#20516 opened Dec 13, 2024 by copybara-service bot Loading…
@majnemer
[XLA:CPU] Add ability to dump the LLvmIrSource to a string in the testlib
#20515 opened Dec 13, 2024 by copybara-service bot Loading…
@WillFroom
[XLA:CPU] Move CPU ElementalIrEmitter implementation into a separate file
#20514 opened Dec 13, 2024 by copybara-service bot Loading…
@WillFroom
PR #19649: [ROCm] Implement hermetic rocm dependency
#20512 opened Dec 13, 2024 by copybara-service bot Loading…
[XLA:CPU] Export codegen testlib functionality via __init__.py
#20510 opened Dec 13, 2024 by copybara-service bot Loading…
@WillFroom
ProTip! Filter pull requests by the default branch with base:main.