Commits

Branch selector

 main

User selector

superbobry

Datepicker

All time

Commit History

Commits on Mar 22, 2024

• Really skip exp2 in Pallas GPU tests with older jaxlib

  Show description for 2d65571
  

  superbobry

  authored and

  jax authors

  committed
  2d65571

  Copy full SHA for 2d65571

Commits on Mar 21, 2024

• Skip PallasOpsTest.test_elementwise_exp2 on older jaxlib

  Show description for c945766
  

  superbobry

  authored and

  jax authors

  committed
  c945766

  Copy full SHA for c945766

• Complemented libdevice ops with the ones from the MLIR Math dialect

  Show description for c5fa14b
  

  superbobry

  authored and

  jax authors

  committed
  c5fa14b

  Copy full SHA for c5fa14b

Commits on Mar 20, 2024

• Use jnp.issubdtype instead of np.issubdtype

  Show description for 9862236
  

  superbobry

  authored and

  jax authors

  committed
  9862236

  Copy full SHA for 9862236

• Added missing BUILD dependencies for Pallas GPU lowering

  Show description for 089651f
  

  superbobry

  authored and

  jax authors

  committed
  089651f

  Copy full SHA for 089651f

Commits on Mar 18, 2024

• List all available backends when jax.devices() is called with an unknown platform

  Show description for 5a2fb38
  

  superbobry

  authored and

  jax authors

  committed
  5a2fb38

  Copy full SHA for 5a2fb38

Commits on Mar 15, 2024

• Update the lowering for div_p to require f32/f64 for floating point inputs

  Show description for cdafb8f
  

  superbobry

  authored and

  jax authors

  committed
  cdafb8f

  Copy full SHA for cdafb8f

Commits on Mar 13, 2024

• Log the exception if the callback passed to jax.*_callback raises

  Show description for a8e2ee9
  

  superbobry

  authored and

  jax authors

  committed
  a8e2ee9

  Copy full SHA for a8e2ee9

• Added a Pallas GPU test for jnp.invert

  Show description for f0c5051
  

  superbobry

  authored and

  jax authors

  committed
  f0c5051

  Copy full SHA for f0c5051

• Removed unused _pack_indices

  Show description for 926e673
  

  superbobry

  authored and

  jax authors

  committed
  926e673

  Copy full SHA for 926e673

• Use int.bit_length() in next_power_of_2

  Show description for 168f30a
  

  superbobry

  authored and

  jax authors

  committed
  168f30a

  Copy full SHA for 168f30a

• Fixed the signature of the fallback get_compute_capability

  Show description for a796444
  

  superbobry

  authored and

  jax authors

  committed
  a796444

  Copy full SHA for a796444

Commits on Mar 12, 2024

• Added tests for more Pallas GPU lowering rules

  Show description for 75dbd30
  

  superbobry

  authored and

  jax authors

  committed
  75dbd30

  Copy full SHA for 75dbd30

• Added a lowering rule for lax.sign_p and improved test coverage for binary ops

  Show description for ae1cfc2
  

  superbobry

  authored and

  jax authors

  committed
  ae1cfc2

  Copy full SHA for ae1cfc2

Commits on Mar 11, 2024

• Fixed lowering of binary ops for signed dtypes

  Show description for 0bd7070
  

  superbobry

  authored and

  jax authors

  committed
  0bd7070

  Copy full SHA for 0bd7070

• Add a deprecation warning to the xmap function and the corresponding tutorial

  superbobry

  committed
  88853c1

  Copy full SHA for 88853c1

• Removed inspect.signature() call from jaxlib.triton.dialect.ScanOp

  Show description for 778933d
  

  superbobry

  authored and

  jax authors

  committed
  778933d

  Copy full SHA for 778933d

• Include source info as ir.Locations when lowering Pallas kernels on GPU

  Show description for 7863508
  

  superbobry

  authored and

  jax authors

  committed
  7863508

  Copy full SHA for 7863508

Commits on Mar 9, 2024

• Deprecated the jax.experimental.maps submodule

  Show description for 930aaa5
  

  superbobry

  authored and

  jax authors

  committed
  930aaa5

  Copy full SHA for 930aaa5

Commits on Mar 7, 2024

• Added a missing cast in _compute_pointers_from_indices

  Show description for c94ec17
  

  superbobry

  authored and

  jax authors

  committed
  c94ec17

  Copy full SHA for c94ec17

Commits on Mar 6, 2024

• Ensured that Pallas GPU tests only run in x32 mode

  Show description for d0e0ca1
  

  superbobry

  authored and

  jax authors

  committed
  d0e0ca1

  Copy full SHA for d0e0ca1

Commits on Mar 4, 2024

• Axis names are now tracked via an effect

  Show description for 5283d4b
  

  superbobry

  authored and

  jax authors

  committed
  5283d4b

  Copy full SHA for 5283d4b

Commits on Feb 27, 2024

• Fixed a typo in the lowering rule for lax.dot_general_p

  Show description for 98e4b9e
  

  superbobry

  authored and

  jax authors

  committed
  98e4b9e

  Copy full SHA for 98e4b9e

• The new Pallas GPU lowering now supports triton_params=

  Show description for 4513a51
  

  superbobry

  authored and

  jax authors

  committed
  4513a51

  Copy full SHA for 4513a51

Commits on Feb 26, 2024

• Moved Pallas GPU lowering registartion code into a separate submodule

  Show description for ad26ba8
  

  superbobry

  authored and

  jax authors

  committed
  ad26ba8

  Copy full SHA for ad26ba8

Commits on Feb 22, 2024

• Use x*x instead of x**2 as an expectation in PallasOpsTest.test_pow_weak_dtype

  Show description for 3b1621b
  

  superbobry

  authored and

  jax authors

  committed
  3b1621b

  Copy full SHA for 3b1621b

• Handle TypeError due to | in type annotations in Triton MLIR bindings

  Show description for d434ab5
  

  superbobry

  authored and

  jax authors

  committed
  d434ab5

  Copy full SHA for d434ab5

• Compile Triton kernels via XLA by default

  Show description for 0bf8ddd
  

  superbobry

  authored and

  jax authors

  committed
  0bf8ddd

  Copy full SHA for 0bf8ddd

Commits on Feb 20, 2024

• Removed deprecated jax.config methods and jax.config.config

  Show description for 57e59eb
  

  superbobry

  authored and

  jax authors

  committed
  57e59eb

  Copy full SHA for 57e59eb

• Fixed internal CI builds

  Show description for 37f313a
  

  superbobry

  authored and

  jax authors

  committed
  37f313a

  Copy full SHA for 37f313a

Commits on Feb 19, 2024

• Allow dynamic grid on GPU in interpret mode

  Show description for 22a8da5
  

  superbobry

  authored and

  jax authors

  committed
  22a8da5

  Copy full SHA for 22a8da5

• Added a few missing compute capability checks to Pallas:GPU tests

  Show description for 46ec581
  

  superbobry

  authored and

  jax authors

  committed
  46ec581

  Copy full SHA for 46ec581

• Generate per-backend tests in //tests/pallas

  Show description for 2101725
  

  superbobry

  authored and

  jax authors

  committed
  2101725

  Copy full SHA for 2101725

• Fix an AttributeError when importing jax.experiemntal.pallas with CPU jaxlib

  Show description for 07d6b81
  

  superbobry

  authored and

  jax authors

  committed
  07d6b81

  Copy full SHA for 07d6b81

• Removed deprecated jax.config methods and jax.config.config

  Show description for b506fee
  

  superbobry

  authored and

  jax authors

  committed
  b506fee

  Copy full SHA for b506fee

Pagination

Previous

Next