Commits

Branch selector

 main

User selector

hawkinsp

Datepicker

All time

Commit History

Commits on Feb 15, 2024

• Micro-optimization: use .dtype rather than dtypes.result_type when forming MLIR ndarray constants.

  Show description for 4834423
  

  hawkinsp

  authored and

  jax authors

  committed
  4834423

  Copy full SHA for 4834423

• Don't call inspect.signature() each time we trace a jit().

  Show description for b5e4ba4
  

  hawkinsp

  authored and

  jax authors

  committed
  b5e4ba4

  Copy full SHA for b5e4ba4

• Don't recompute abstract eval rules when inlining a jit jaxpr.

  Show description for 885e8a2
  

  hawkinsp

  authored and

  jax authors

  committed
  885e8a2

  Copy full SHA for 885e8a2

• Remove code that guards against int4 not being available.

  Show description for c55f187
  

  hawkinsp

  authored and

  jax authors

  committed
  c55f187

  Copy full SHA for c55f187

Commits on Feb 14, 2024

• Summarize equations in sharding mismatch errors lazily.

  Show description for 9bb6f18
  

  hawkinsp

  authored and

  jax authors

  committed
  9bb6f18

  Copy full SHA for 9bb6f18

• Don't compute the log message for checkpointing residuals unless it is going to be logged.

  Show description for 7156f20
  

  hawkinsp

  authored and

  jax authors

  committed
  7156f20

  Copy full SHA for 7156f20

Commits on Feb 13, 2024

• Speed up user filename identification.

  Show description for cfcae37
  

  hawkinsp

  authored and

  jax authors

  committed
  cfcae37

  Copy full SHA for cfcae37

• Move test for float0 values out of nary_op_weak_type_rule and into unop_dtype_rule.

  Show description for 66a4dc5
  

  hawkinsp

  authored and

  jax authors

  committed
  66a4dc5

  Copy full SHA for 66a4dc5

• Micro optimization to standard_abstract_eval:

  Show description for bab8ae8
  

  hawkinsp

  authored and

  jax authors

  committed
  bab8ae8

  Copy full SHA for bab8ae8

• Use an isinstance check rather than dtypes.issubdtype to check whether the dtype in an aval is an extended dtype.

  Show description for 5833b07
  

  hawkinsp

  authored and

  jax authors

  committed
  5833b07

  Copy full SHA for 5833b07

• Don't look for dimension_as_value on Tracers in core.full_raise.

  Show description for 70edc40
  

  hawkinsp

  authored and

  jax authors

  committed
  70edc40

  Copy full SHA for 70edc40

• Small optimizations to mlir._traceback_to_location.

  Show description for 2a8634b
  

  hawkinsp

  authored and

  jax authors

  committed
  2a8634b

  Copy full SHA for 2a8634b

• Make config.value_holders private to avoid misuse.

  Show description for 86c8af7
  

  hawkinsp

  authored and

  jax authors

  committed
  86c8af7

  Copy full SHA for 86c8af7

• Reverts 8e96c49f85321a4df9549a9f3f813bb7eb71b139

  Show description for 8b1da12
  

  hawkinsp

  authored and

  jax authors

  committed
  8b1da12

  Copy full SHA for 8b1da12

• Fix lint failure in fused attention test.

  Show description for 7aadabd
  

  hawkinsp

  authored and

  jax authors

  committed
  7aadabd

  Copy full SHA for 7aadabd

Commits on Feb 12, 2024

• Small optimization to trace_context.

  Show description for 995bdaa
  

  hawkinsp

  authored and

  jax authors

  committed
  995bdaa

  Copy full SHA for 995bdaa

• Refactor and optimize the implementation of config options.

  Show description for c1f234a
  

  hawkinsp

  authored and

  jax authors

  committed
  c1f234a

  Copy full SHA for c1f234a

• Tighten up typing of optional config values.

  Show description for 6b8c99c
  

  hawkinsp

  authored and

  jax authors

  committed
  6b8c99c

  Copy full SHA for 6b8c99c

Commits on Jan 22, 2024

• [XLA:Python] Fail with an AttributeError if __cuda_array_interface__ is called on a sharded array.

  Show description for dfda948
  

  hawkinsp

  authored and

  jax authors

  committed
  dfda948

  Copy full SHA for dfda948

• Disable NumPy interoperability DLPack test for bools for NumPy versions older than 1.25.

  Show description for 71ee3ab
  

  hawkinsp

  authored and

  jax authors

  committed
  71ee3ab

  Copy full SHA for 71ee3ab

• Disabled lax_scipy_special_functions_test under ASAN on GPU.

  Show description for 25e4acf
  

  hawkinsp

  authored and

  jax authors

  committed
  25e4acf

  Copy full SHA for 25e4acf

Commits on Jan 19, 2024

• [JAX] Disable a compilation cache test that fails on Windows in CI.

  Show description for a7023b1
  

  hawkinsp

  authored and

  jax authors

  committed
  a7023b1

  Copy full SHA for a7023b1

• Add a new experimental option jax_pmap_no_rank_reduction.

  Show description for fc6df32
  

  hawkinsp

  authored and

  jax authors

  committed
  fc6df32

  Copy full SHA for fc6df32

Commits on Jan 17, 2024

• Add support for bool dlpack values.

  Show description for c436835
  

  hawkinsp

  authored and

  jax authors

  committed
  c436835

  Copy full SHA for c436835

Commits on Jan 15, 2024

• Deprecate support for the mhlo dialect.

  Show description for e558fea
  

  hawkinsp

  authored and

  jax authors

  committed
  e558fea

  Copy full SHA for e558fea

• Disable some tests that time out or OOM under ASAN.

  Show description for 912a5ef
  

  hawkinsp

  authored and

  jax authors

  committed
  912a5ef

  Copy full SHA for 912a5ef

Commits on Jan 12, 2024

• [Pallas] Disable sanitizer builds of splash_attention_kernel_test.

  Show description for d935213
  

  hawkinsp

  authored and

  jax authors

  committed
  d935213

  Copy full SHA for d935213

• [JAX] Disable some CUDA tests that fail under ASAN, due to bugs in NCCL and Triton.

  Show description for 2980e8f
  

  hawkinsp

  authored and

  jax authors

  committed
  2980e8f

  Copy full SHA for 2980e8f

Commits on Jan 11, 2024

• Add a bazel test that verifies that the jaxlib wheel builds.

  hawkinsp

  committed
  dedd69f

  Copy full SHA for dedd69f

• Disable ASAN for several CUDA tests.

  Show description for 35fc2ed
  

  hawkinsp

  authored and

  jax authors

  committed
  35fc2ed

  Copy full SHA for 35fc2ed

• Fix jaxlib wheel build after removal of mosaic python files.

  Show description for 858fd52
  

  hawkinsp

  authored and

  jax authors

  committed
  858fd52

  Copy full SHA for 858fd52

Commits on Jan 4, 2024

• Fix/disable two tests failing in Windows CI.

  Show description for 8803774
  

  hawkinsp

  authored and

  jax authors

  committed
  8803774

  Copy full SHA for 8803774

• [XLA:Python] Raise an AttributeError if __cuda_array_interface__ is called on various invalid buffers, rather than a RuntimeError.

  Show description for 3ff4eb4
  

  hawkinsp

  authored and

  jax authors

  committed
  3ff4eb4

  Copy full SHA for 3ff4eb4

• Guard call to os.register_at_fork() to unbreak Windows CI.

  Show description for 05a9906
  

  hawkinsp

  authored and

  jax authors

  committed
  05a9906

  Copy full SHA for 05a9906

Commits on Dec 19, 2023

• [JAX:GPU] Add a test that verifies that the XLA_PYTHON_CLIENT_PREALLOCATE environment variable is parsed correctly.

  Show description for 67d5c3b
  

  hawkinsp

  authored and

  jax authors

  committed
  67d5c3b

  Copy full SHA for 67d5c3b

Pagination

Previous

Next