ROCm/jax

Pull Requests Commits

correct pytest_abort_plugin install

gulsumgudukbay committed 1 day ago

7cc76ae6

update permissions for bash script

gulsumgudukbay committed 1 day ago

79515e6b

Add abort input to nightly yml file

gulsumgudukbay committed 1 day ago

6934b3e5

adding call to run_pytest_rocm_abort.sh in the workflow yml file

gulsumgudukbay committed 1 day ago

8d084a2f

added run_pytest_rocm_abort.sh script, same as run_pytest_rocm, but with abort detection

gulsumgudukbay committed 1 day ago

94e74333

[TESTING ONLY] Bypass ROCm plugin version check and strip nightly workflow

mminutoli committed 1 day ago

9b8c026b

Expand ROCm testing matrix and improve job naming

mminutoli committed 2 days ago

f9d2e1b5

Add JAXCI_ROCM_VERSION environment variable

mminutoli committed 2 days ago

6ad760e6

Add ROCm pytest workflow to nightly/release CI

mminutoli committed 2 days ago

f952348c

Add pytest_rocm.yml workflow for ROCm testing

mminutoli committed 2 days ago

62f1e56c

Add download-jax-rocm-wheels composite action

mminutoli committed 2 days ago

7f3105e7

Add ROCm wheel patterns to install_wheels_locally.sh

mminutoli committed 2 days ago

1cad7000

Complete run_pytest_rocm.sh for CI integration

mminutoli committed 2 days ago

be9a2c2f

adding run_pytest_rocm.sh

gulsumgudukbay committed 2 days ago

9c629b47

Propagate memory spaces from inputs to outputs when aliased.

brianwa84 committed 2 days ago

5f5be6f1

Switch all Windows RBE builds to use `ml-oss-rbe-general`

nitins17 committed 2 days ago

41256530

Save CPU version of collective metadata to resources so it can be used during execution.

Google-ML-Automation committed 2 days ago

b2c727e3

Merge pull request #34770 from ROCm:add_support_of_building_jax_under_umbrella_workspace_new2

Google-ML-Automation committed 2 days ago

68992a18

Address review comments

alekstheod committed 3 days ago

60f50910

[mosaic] Added a new tpu.fetch_and_add_sync op

superbobry committed 3 days ago

2f1645f1

Add support of executing jax tests from withing a parent workspace as @jax//tests:gpu_tests

alekstheod committed 3 days ago

034c430c

Add some more information to deprecation messages in pxla modules

yashk2810 committed 3 days ago

0775e2b3

Reverts 4a5ad80ec771ec08bea2b28169d2f298aecbb46e

hawkinsp committed 3 days ago

5c4c7f05

Merge pull request #34682 from olupton:workaround-tcgen05-llvm-errors

Google-ML-Automation committed 3 days ago

8288263b

[Mosaic GPU] Clean up all the linter warnings in `dialect_lowering.py`.

bchetioui committed 3 days ago

40c3359e

[Pallas/Mosaic GPU] Remove unnecessary jaxlib version guard-related code.

bchetioui committed 3 days ago

a5854e44

Update XLA dependency to use revision http://github.com/openxla/xla/commit/80529bdfc1c3bc206d0809a474031e9c9008c9f0

Google-ML-Automation committed 3 days ago

1066aa7b

[mosaic] Added a macro which defines AbslStringify for all MLIR enums in the enclosing namespace

superbobry committed 3 days ago

150a6d9d

[mosaic] Avoid using the deprecated `output_memory_colors` in `CustomCallBackendConfig`

superbobry committed 4 days ago

79fd127f

Regenerate python stubs for nanobind 2.11.0.

hawkinsp committed 4 days ago

6fb3f3da

Older