Commits

Branch selector

 master

User selector

skye

Datepicker

All time

Commit History

Commits on Feb 21, 2023

• Add back opt-barrier fallback, since the fallback sometimes prevents OOMs.

  Show description for 6572fac
  skye

  committed
  6572fac

  Copy full SHA for 6572fac

• Add --pre to nightly libtpu pip install command.

  Show description for 1476a85
  skye

  committed
  1476a85

  Copy full SHA for 1476a85

Commits on Feb 16, 2023

• MeshComputation.cost_analysis() isn't implemented with PJRT C API.

  Show description for c2819cf
  skye

  committed
  c2819cf

  Copy full SHA for c2819cf

• Raise a user-friendly error message if in/outfeed-based host_callback stuff is used with PJRT C API.

  Show description for d9f628c
  skye

  committed
  d9f628c

  Copy full SHA for d9f628c

Commits on Feb 15, 2023

• Modify JaxArrayTest.test_defragment to work on any numbers of devices

  Show description for 7aa7e15
  

  skye

  authored and

  jax authors

  committed
  7aa7e15

  Copy full SHA for 7aa7e15

Commits on Feb 11, 2023

• Add back loading TPU plugin for older jaxlib versions.

  Show description for e548585
  

  skye

  authored and

  jax authors

  committed
  e548585

  Copy full SHA for e548585

Commits on Feb 8, 2023

• Post 0.4.3 release updates

  skye

  committed
  21f1218

  Copy full SHA for 21f1218

• Change the executable_build_options check in compilation_cache.py to be more robust.

  Show description for 1228cbd
  skye

  committed
  1228cbd

  Copy full SHA for 1228cbd

Commits on Feb 7, 2023

• Update WORKSPACE and setup.py for jax/jaxlib 0.4.3 release

  skye

  committed
  8ab1585

  Copy full SHA for 8ab1585

• Add option to run tests with persistent compilation cache enabled.

  Show description for eb13c05
  

  skye

  authored and

  jax authors

  committed
  eb13c05

  Copy full SHA for eb13c05

• Don't write executables with host callbacks to persistent compilation cache.

  Show description for 6cef087
  

  skye

  authored and

  jax authors

  committed
  6cef087

  Copy full SHA for 6cef087

• Correctly hash auto_spmd fields in compilation cache key.

  Show description for 2eb10d2
  

  skye

  authored and

  jax authors

  committed
  2eb10d2

  Copy full SHA for 2eb10d2

Commits on Jan 27, 2023

• Add PJRT C API to Cloud TPU test matrix

  Show description for 93cd07e
  skye

  committed
  93cd07e

  Copy full SHA for 93cd07e

Commits on Jan 26, 2023

• Add warning filter to ArrayPjitTest.test_pmap_pjit_axis_index

  skye

  committed
  49e751b

  Copy full SHA for 49e751b

• Update versions for jaxlib 0.4.2 release.

  Show description for b6a8aa6
  skye

  committed
  b6a8aa6

  Copy full SHA for b6a8aa6

Commits on Jan 25, 2023

• [TPU CI] Send chat notification on cancellation as well as failure.

  Show description for 5825782
  skye

  committed
  5825782

  Copy full SHA for 5825782

• Update libtpu version for jaxlib 0.4.2 release (again)

  skye

  committed
  c4ad27c

  Copy full SHA for c4ad27c

Commits on Jan 24, 2023

• Update WORKSPACE for jaxlib 0.4.2 release (again)

  Show description for 9a7f29a
  skye

  committed
  9a7f29a

  Copy full SHA for 9a7f29a

Commits on Jan 20, 2023

• Updates for jax + jaxlib 0.4.2 release

  skye

  committed
  3f4bd5f

  Copy full SHA for 3f4bd5f

• Disable timing out sparse_test.py on msan

  Show description for 953910a
  

  skye

  authored and

  jax authors

  committed
  953910a

  Copy full SHA for 953910a

• Increase sharding on checkify_test.py to avoid asan timeouts

  Show description for 068423b
  

  skye

  authored and

  jax authors

  committed
  068423b

  Copy full SHA for 068423b

Commits on Jan 18, 2023

• Don't run FP8 dtype test on TPU.

  Show description for 6d0e22e
  

  skye

  authored and

  jax authors

  committed
  6d0e22e

  Copy full SHA for 6d0e22e

Commits on Jan 13, 2023

• Don't set TPU topology env vars in cloud_tpu_init.py

  Show description for 004b8c1
  skye

  committed
  004b8c1

  Copy full SHA for 004b8c1

Commits on Jan 12, 2023

• Migrate pytestmark usage to new @jtu.pytest_mark_if_available decorator.

  Show description for c0577f7
  skye

  committed
  c0577f7

  Copy full SHA for c0577f7

• Add pjrt_c_api_unimplemented pytest marker to skip unsupported tests.

  Show description for f90b5ee
  skye

  committed
  f90b5ee

  Copy full SHA for f90b5ee

Commits on Dec 9, 2022

• [TPU CI] Run build matrix on v3-8 as well as v4-8

  Show description for 8d4b50e
  skye

  committed
  8d4b50e

  Copy full SHA for 8d4b50e

Commits on Dec 1, 2022

• [docs] Replace one more jax_Array.html reference

  Show description for 82b442f
  skye

  committed
  82b442f

  Copy full SHA for 82b442f

• [docs] Rename "JAX in Parallelism" files so the URL matches the title.

  skye

  committed
  51db1cf

  Copy full SHA for 51db1cf

Commits on Nov 23, 2022

• Fix some logging to use logger instead of logging

  skye

  committed
  6d79e2f

  Copy full SHA for 6d79e2f

• Add --maxfail=20 to Cloud TPU CI.

  Show description for 246614e
  skye

  committed
  246614e

  Copy full SHA for 246614e

Commits on Nov 18, 2022

• Make pytest-xdist work on TPU and update Cloud TPU CI.

  Show description for 120125f
  skye

  committed
  120125f

  Copy full SHA for 120125f

Commits on Nov 16, 2022

• Include which jaxlib/libtpu version failed (latest or nightly) in TPU CI chat notification

  skye

  committed
  0a886c3

  Copy full SHA for 0a886c3

• TPU CI: don't notify when testing the workflow from a branch

  skye

  committed
  b4564a2

  Copy full SHA for b4564a2

• Update Github Actions workflows using Ratchet

  Show description for 8bed9ba
  skye

  committed
  8bed9ba

  Copy full SHA for 8bed9ba

• Update TensorBoard install instructions for profiling

  skye

  committed
  66f3e0d

  Copy full SHA for 66f3e0d

Pagination

Previous

Next