ROCm/jax

Pull Requests Commits

Update sanity.yml

okakarpa committed 1 year ago

Verified 621f957f

Update sanity.yml

okakarpa committed 1 year ago

Verified 1f978fcd

Update arc-runner.yml

okakarpa committed 1 year ago

Verified aedbbb1e

Update sanity.yml

okakarpa committed 1 year ago

Verified 6bf5af87

Update sanity.yml

okakarpa committed 1 year ago

Verified de4e6780

Create arc-runner.yml

okakarpa committed 1 year ago

Verified e9c5b2d0

Update sanity.yml

okakarpa committed 1 year ago

Verified ff896655

Update and rename rocm-sanity.yml to sanity.yml

okakarpa committed 1 year ago

Verified 994ad73e

Update rocm-sanity.yml

okakarpa committed 1 year ago

Verified a2b1bd2f

Update rocm-sanity.yml

okakarpa committed 1 year ago

Verified e6bdaff0

Update and rename pylint.yml to rocm-sanity.yml

okakarpa committed 1 year ago

Verified dbf5c1c8

Update pylint.yml

okakarpa committed 1 year ago

Verified 854d6be7

Create pylint.yml

okakarpa committed 1 year ago

Verified 1e115020

[ROCm] improve gpu script

Ruturaj4 committed 2 years ago

644ac10c

[ROCM] Fix BUILD.bazel library source paths

Ruturaj4 committed 2 years ago

a2d79936

Add `mesh_shape` to the lowering context. This is to allow custom partitioning to not depend on the mesh context manager to return NamedShardings even if the arguments have NamedShardings on them.

yashk2810 committed 2 years ago

dd958adc

Change log level of clearing JAX backend caches from info to debug.

yueshengys committed 2 years ago

7f44edc0

Update XLA dependency to use revision

Google-ML-Automation committed 2 years ago

798297af

Merge pull request #22900 from jakevdp:dep-bfloat16

a-googler committed 2 years ago

53ab5eb2

Add test for zero-sized host memory parameter

Google-ML-Automation committed 2 years ago

9074e854

Merge pull request #22649 from ROCm:ci_jax_export_harness

a-googler committed 2 years ago

aec6efb4

Merge pull request #22901 from ROCm:ci_test_harness_vmap

a-googler committed 2 years ago

cc966574

Remove enable_gpu and xla_python_enable_gpu from jax .bazelrc.

Jieying Luo committed 2 years ago

abe7982d

Skip flaky test_weight_offload_with_dp_on_output test on GPU backend.

kanglant committed 2 years ago

ae541203

[ROCM] Fix hipsolverSsyevd tests due to align with the rocm behavior.

Ruturaj4 committed 2 years ago

707cdd47

deprecate jax.lib.xla_client.bfloat16

jakevdp committed 2 years ago

a009e1cf

Merge pull request #22834 from rajasekharporeddy:testbranch1

a-googler committed 2 years ago

f67f73c3

Merge pull request #22896 from jakevdp:pin-sphinx

a-googler committed 2 years ago

799de71c

CI: pin sphinx to avoid build errors on 8.0

jakevdp committed 2 years ago

4f8c5a33

[ROCM] Fix export harness tests

Ruturaj4 committed 2 years ago

35c70fd3

Older