ROCm/jax

Pull Requests Commits

Trigger CI/CD pipelinee

alekstheod committed 12 days ago

0c856ce1

Address review comments

alekstheod committed 13 days ago

d21547af

Adress review comments

alekstheod committed 13 days ago

d151959c

alekstheod committed 13 days ago

4441a183

alekstheod committed 13 days ago

d583f1a0

Address review comments, use repo name as a convention

alekstheod committed 13 days ago

a58e5e5d

Address review comments

alekstheod committed 13 days ago

40462512

Address review comments, do not use env variables

alekstheod committed 13 days ago

6cb71238

Use Label() instead

alekstheod committed 13 days ago

b2be9cca

Add valid repo resolution name for making it work as an external project

alekstheod committed 13 days ago

0a2f4cca

Adjust name to external_deps_repository

alekstheod committed 13 days ago

a50a4dec

Rename test_deps to external_deps.bzl

alekstheod committed 13 days ago

d02fcbae

Move deps to gpu_test_deps

alekstheod committed 13 days ago

e615e898

alekstheod committed 13 days ago

87707e6e

Address review comments

alekstheod committed 13 days ago

a8b2837d

Address review comments

alekstheod committed 13 days ago

f7894478

Address review comments

alekstheod committed 14 days ago

2b34afd5

Add missing bzlmod depenency

alekstheod committed 15 days ago

c6ad61b8

Add support to dynamically set test deps for jax as an external repo

alekstheod committed 15 days ago

1f4c34f2

[mosaic] Added a new tpu.fetch_and_add_sync op

superbobry committed 15 days ago

2f1645f1

Add some more information to deprecation messages in pxla modules

yashk2810 committed 15 days ago

0775e2b3

Reverts 4a5ad80ec771ec08bea2b28169d2f298aecbb46e

hawkinsp committed 15 days ago

5c4c7f05

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

Google-ML-Automation committed 15 days ago

8288263b

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

bchetioui committed 15 days ago

40c3359e

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

bchetioui committed 15 days ago

a5854e44

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

Google-ML-Automation committed 15 days ago

1066aa7b

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

superbobry committed 16 days ago

150a6d9d

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

superbobry committed 16 days ago

79fd127f

Regenerate python stubs for nanobind 2.11.0.

hawkinsp committed 16 days ago

6fb3f3da

Update XLA dependency to use revision http://github.com/openxla/xla/commit/2a7188001603266f57a661a10beee3418412c312

Google-ML-Automation committed 16 days ago

3a93274b

Older