ROCm/jax

Pull Requests Commits

fixed ci script for ubuntu distro

zahiqbal committed 1 year ago

334e1bb7

Add git fetch (#132)

charleshofer committed 1 year ago

Verified 350e04d8

Fix upstream sync checkout (#130)

charleshofer committed 1 year ago

Verified fbd409db

Create a new branch when merging upstream main to rocm-main (#128)

charleshofer committed 1 year ago

Verified 69e93e5a

Add GHA workflow for opening PRs upstream (#116)

charleshofer committed 1 year ago

Verified ea7683f0

Fix nightly sync permissions (#124)

charleshofer committed 1 year ago

Verified 650e70ab

Rename the CI flow to 'ROCm CI' and only run it on PRs to rocm-main (#126)

charleshofer committed 1 year ago

Verified 700f3bdc

Merge pull request #123 from ROCm/gh-9948-fix-ci-build-image

JehandadKhan committed 1 year ago

Verified 4dd55ffb

Unpin container in CI build and remove libssl-dev install

charleshofer committed 1 year ago

14139a3f

Allow devs to kick off sync job manually (#119)

charleshofer committed 1 year ago

Verified 7a152655

Merge pull request #115 from ROCm/gh-9948-upstream-sync

charleshofer committed 1 year ago

Verified 2f02da61

Set runners for ROCM

charleshofer committed 1 year ago

1f66c29d

Fix yaml checker

charleshofer committed 1 year ago

3361fca5

Only run on weekdays

charleshofer committed 1 year ago

1c877f5a

Add workflow for nightly pull from upstream

charleshofer committed 1 year ago

692e2256

[Mosaic] Add the core type enum

naummo committed 1 year ago

242e6634

Update XLA dependency to use revision

Google-ML-Automation committed 1 year ago

af14c438

#sdy add shardy CPU config for all JAX tests, disabling any known failing test cases.

bartchr808 committed 1 year ago

44158ab0

Add a temporary fix for spurious debug_nans errors when round-tripping jaxprs.

dougalm committed 1 year ago

32bf19ac

[Mosaic] Test only cl - add triu test, skip bf16 due to select being native bitwidth only

Google-ML-Automation committed 1 year ago

3904ced2

[Fix] Put * packing into reshape no-op condition (Bug in my original CL)

Google-ML-Automation committed 1 year ago

99ea4c1a

[pallas:mosaic_gpu] Disabled verbose lowering errors in Mosaic GPU tests

superbobry committed 1 year ago

409517fc

[pallas] Added a flag disabling verbose error reporting

superbobry committed 1 year ago

6283eab2

Move utility functions in build.py to utils.py

nitins17 committed 1 year ago

da994d35

[Pallas] Add test cases for var + constant.

WindQAQ committed 1 year ago

d2f58044

Fix a CI failure under NumPy 2.1.

hawkinsp committed 1 year ago

a8f44c47

Reverts 6dd1417d4a0a9ee31d8a014352b3a0fb2bcfcbaf

tkoeppe committed 1 year ago

2bed1e88

[mosaic_gpu] Added support for bitwise and, or and xor to `FragmentedArray`

superbobry committed 1 year ago

2652ab56

Removed unused `_upcast_fp16_for_computation`

superbobry committed 1 year ago

2b70ad30

Relax leak checks under the jax_data_dependent_tracing_fallback flag.

dougalm committed 1 year ago

a45b0856

Older