ROCm/jax

Pull Requests Commits

test: skip complex geomspace test under numpy 2.0

jakevdp committed 1 year ago

c79f54f7

This CL introduces 'PluginProgram' in IFRT and exposes this in python via `xla_client.compile_ifrt_program()`.

a-googler committed 1 year ago

fb55d591

Fail gracefully in lobpcg_test if matplotlib isn't installed.

hawkinsp committed 1 year ago

60458bb3

Fix a bug where exceptions were thrown in debug message formatting, when sharding was set to None on arrays.

a-googler committed 1 year ago

00489be2

jnp.broadcast_to: better error for invalid shape

jakevdp committed 1 year ago

6de6983d

Added installation instructions to the error in _pallas_call_lowering

superbobry committed 1 year ago

2ee4c0f6

Merge pull request #20514 from gnecula:callback_cache

a-googler committed 1 year ago

4c41c12e

[callback] Improve caching effectiveness in presence of callbacks.

gnecula committed 1 year ago

bff24c6d

Merge pull request #20383 from gnecula:doc_deprecation

a-googler committed 1 year ago

431015a1

Add `round` lowering rule.

a-googler committed 1 year ago

b3fe9400

A few more comments about how the deprecations work

gnecula committed 1 year ago

84db689e

Update XLA dependency to use revision

a-googler committed 1 year ago

1d221d1f

Accelerate deprecation of jax.experimental.host_callback.id_print and stop_outfeed_receiver

gnecula committed 1 year ago

c491720e

Register GPU/TPU lowering for pallas_call_p lazily

superbobry committed 1 year ago

16b3f00e

Merge pull request #20524 from jakevdp:trapz

a-googler committed 1 year ago

5a7e8743

Generalized the in_specs/out_specs types in PrefetchScalarGridSpec

superbobry committed 1 year ago

c4f1a452

Rename SpecifiedLayout to DeviceLocalLayout

yashk2810 committed 1 year ago

6557f680

Add jax.numpy.trapezoid

jakevdp committed 1 year ago

9e01afe7

Guard test that requires two devices with device_count() check.

hawkinsp committed 1 year ago

011ced44

Merge pull request #20520 from google:dependabot/github_actions/actions/setup-python-5.1.0

a-googler committed 1 year ago

a4cccd98

Merge pull request #20474 from rajasekharporeddy:test_branch3

a-googler committed 1 year ago

5ece588f

Bump actions/setup-python from 5.0.0 to 5.1.0

dependabot[bot] committed 1 year ago

Verified 7a57b39a

Merge pull request #20426 from google:dependabot/github_actions/actions/cache-4.0.2

a-googler committed 1 year ago

3e9bb51d

Merge pull request #20454 from pearu:pearu/log1p-disable-tests

a-googler committed 1 year ago

29cb8955

[PJRT:CPU] Fix thread-pool stack sizes to 2MiB.

hawkinsp committed 1 year ago

77ff8a23

Disable asan for paged_attention_kernel_test.

voutcn committed 1 year ago

7f9ff82e

Update XLA dependency to use revision

a-googler committed 1 year ago

3f3986fd

Update XLA dependency to use revision

a-googler committed 1 year ago

81c3d51d

Update XLA dependency to use revision

a-googler committed 1 year ago

efa37cf3

[PJRT C API] Add a PJRT extension to register custom partitioner.

Jieying Luo committed 1 year ago

68c674d1

Newer Older