Commits ROCm/jax - SemanticDiff

Merge pull request #34986 from gnecula:fix_test_error

Google-ML-Automation committed 10 days ago

0d3230b1

Merge pull request #34689 from ROCm:rocm_fix_pallas

Google-ML-Automation committed 10 days ago

ebbdfdce

Fix failing test in array_interoperability_test.py

gnecula committed 10 days ago

3166c321

Some fixes following #34911

superbobry committed 10 days ago

b8723498

Reverts 583744603265ea24dfaca6a1300e46b65377b3ef

h-joo committed 10 days ago

001c1ad2

Merge pull request #34578 from ROCm:ups-select-skip-conv

Google-ML-Automation committed 10 days ago

6b3a3e9b

Merge pull request #34534 from magaonka-amd:enable-lax-dot-rocm-jax-ml

Google-ML-Automation committed 10 days ago

07ef410d

Automated Code Change

h-joo committed 10 days ago

58374460

Merge pull request #34577 from ROCm:rocm-enable-workspace-overflow-test

Google-ML-Automation committed 10 days ago

1202f110

Merge pull request #34567 from ROCm:unskip_multivariate_normal_rocm

Google-ML-Automation committed 10 days ago

fb31d9fd

Merge pull request #34571 from magaonka-amd:enable-reduce-window-tests-rocm-upstream

Google-ML-Automation committed 10 days ago

4c157782

Merge pull request #34890 from magaonka-amd:rocm-enable-memory-space-export-tests

Google-ML-Automation committed 10 days ago

14a1491e

Update XLA dependency to use revision http://github.com/openxla/xla/commit/11b961377ccfb82d0b0256c9852c0042a67e520e

Google-ML-Automation committed 10 days ago

8d9e1809

Use `reshape` in `_reshape_transpose_rule` instead of `reshape_p.bind` as it was before unreduced/reduced support was added to reshape_p.

yashk2810 committed 10 days ago

623d9dba

Merge pull request #34304 from ROCm:rocm_psend_support

Google-ML-Automation committed 10 days ago

d76e2431

Merge pull request #34662 from magaonka-amd:enable-caitojax-rocm

Google-ML-Automation committed 10 days ago

ef18ec70

Merge pull request #34470 from ROCm:rocm_enable_tridiagonal_tests

Google-ML-Automation committed 10 days ago

dc51c478

Merge pull request #34959 from ROCm:skip-tridiagonal-solve-grad-upstream-main

Google-ML-Automation committed 10 days ago

7136c7ab

Merge pull request #34560 from magaonka-amd:enable-optimizer-tests-rocm

Google-ML-Automation committed 10 days ago

cd55b742

Make reshape more restritive over unreduced/reduced. Concretely, `operand.aval.sharding.spec.{unreduced,reduced}` should be equal to `sharding.spec.{unreduced,reduced}`.

yashk2810 committed 10 days ago

64578697

Adding a FREETHREADED_FLAG_VALUE to let us test Python 3.13 manually. Because JAX CI for Continuous Wheel Tests support 3.13, we need this flag to force those tests to run. This ensures the rules_python upgrade doesn't break compatibility with the new Python version during testing of upgrades.

ecalubaquib committed 10 days ago

1d498d63

Merge pull request #34600 from ROCm:phambinh/ci_rocm_scaled_matmul_tests

Google-ML-Automation committed 10 days ago

c2a29f63

Merge pull request #34603 from ROCm:phambinh/ci_rocm_misc_tests

Google-ML-Automation committed 10 days ago

a3ab316e

Fix XLA override in BZLMOD.

ybaturina committed 10 days ago

77c4897d

Add reshape unreduced and reduced rule. `reshape` just propagates unreduced/reduced on operand to output.

yashk2810 committed 10 days ago

f480660e

Merge pull request #34794 from jakevdp:pyrefly-fixes

Google-ML-Automation committed 10 days ago

7bbbc474

Update XLA dependency to use revision http://github.com/openxla/xla/commit/11991b81799e9f54a1eb6df417f089195f207c4f

Google-ML-Automation committed 10 days ago

e07d0645

[Mosaic GPU] Add missing assertion in `test_serialize_uses_bytecode_format`.

bchetioui committed 10 days ago

f1067f9a

Merge pull request #34939 from mattjj:disallow-f16-reductions

Google-ML-Automation committed 11 days ago

92506c17

[typing] some configuration & local fixes for pyrefly

jakevdp committed 11 days ago

db600bd9