ROCm/jax

Pull Requests Commits

Add additional tsan suppressions for races.

hawkinsp committed 1 year ago

d4c66578

Merge pull request #26238 from hawkinsp:coretest

Google-ML-Automation committed 1 year ago

0ef2ccfd

Merge pull request #26239 from jakevdp:rotation-test

Google-ML-Automation committed 1 year ago

b039f976

Merge pull request #26245 from jakevdp:fix-line

Google-ML-Automation committed 1 year ago

70aed64a

Merge pull request #26194 from jax-ml:fix-dist-init-runtime-error

Google-ML-Automation committed 1 year ago

cb79ff4d

delete unnecessary line

jakevdp committed 1 year ago

0df7f182

Merge pull request #26236 from adonath:sphinx-copybutton-no-prompt-copy

Google-ML-Automation committed 1 year ago

def5fd52

[pjrt] Use the `PjRtMemorySpace` version of `BufferFromHostLiteral`

superbobry committed 1 year ago

5cad02c1

Avoid call to asarray in jnp.einsum

jakevdp committed 1 year ago

4e30a08e

Lower `np.ndarray` to `DenseElementsAttr` instead of `ArrayAttr`.

ghpvnist committed 1 year ago

20555f63

Adjust conf.py to remove prompt from sphinx copybutton

adonath committed 1 year ago

dd7c203b

Merge pull request #26220 from jakevdp:lax-docs-1

Google-ML-Automation committed 1 year ago

ccef8301

Fix GCS download URI in Bazel CUDA non-RBE workflow

nitins17 committed 1 year ago

04476cf1

jax.lax: improve documentation for several functions

jakevdp committed 1 year ago

522200ff

test: change random matrix generation for Rotation

jakevdp committed 1 year ago

4d433d06

Disable two tests of GC behavior if there are multiple threads per process.

hawkinsp committed 1 year ago

2d9cd86a

[mgpu] Correct instruction for conversion of unsigned int types.

cperivol committed 1 year ago

bf967173

Re-enable Pallas bf16 exp2 tests on TPU

apaszke committed 1 year ago

c1e13605

Reverts 86643a1b3e0516e1a2ddbdabbb714cf8c0301f18

belitskiy committed 1 year ago

1bfdd504

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

e6e7621f

[CI] Update naming comments in workflow files.

MichaelHudgins committed 1 year ago

873db96f

[Mosaic GPU] Allow uneven partitioning of dimensions into tiles in TileTransform

apaszke committed 1 year ago

cadfcc7a

[Mosaic GPU] Add support for tiled swizzle=16 (i.e. no swizzle) loads and stores

apaszke committed 1 year ago

10ac6b7e

Merge pull request #26218 from jakevdp:bump-array-api

Google-ML-Automation committed 1 year ago

efacec4c

Fix CI failures in xla_metadata_test.

hawkinsp committed 1 year ago

60fad99c

Pass filter=data to tar extractall to avoid a warning under Python 3.12+

hawkinsp committed 1 year ago

0705ec2c

Disable a debug_info_test test that fails in CI.

hawkinsp committed 1 year ago

a2f7824c

[Mosaic TPU] Support non-32 bit mask relayout

bythew3i committed 1 year ago

785a63ad

Merge pull request #26221 from justinjfu:remove_smap_test

Google-ML-Automation committed 1 year ago

9dfe03c5

remove checks since they are redundant and we can change out_aval because of various reasons

yashk2810 committed 1 year ago

1f33cad3

Newer Older