Commits

Breadcrumbs

History for

onmaster

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 19, 2023

• Remove *_mhlo compatibility shims from jaxlib

  Show description for f337c00
  

  Eugene Burmako

  authored and

  jax authors

  committed
  f337c00

Commits on Feb 17, 2023

• Limit visibility of Bazel target jax:global_device_array.

  Show description for f7734fd
  

  hawkinsp

  authored and

  jax authors

  committed
  f7734fd

Commits on Feb 15, 2023

• Add new argument to jax_test rule

  Show description for 936e4ae
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  936e4ae

Commits on Feb 13, 2023

• fix change to csr lowering rule

  jakevdp

  committed
  ddae1d0

Commits on Feb 11, 2023

• [sparse] implement BCSR.__matmul__

  jakevdp

  committed
  de8a77a

Commits on Feb 6, 2023

• Bump minimum jaxlib_version to 0.4.2 i.e xla_extension_version == 119 and mlir_api_version == 43

  Show description for 8a69444
  

  yashk2810

  authored and

  jax authors

  committed
  8a69444

Commits on Dec 27, 2022

• Migrate JAX from producing MHLO to producing StableHLO

  Show description for a1480c4
  

  Eugene Burmako

  authored and

  jax authors

  committed
  a1480c4

• Add support for dynamic shapes to GPU threefry2x32 custom call.

  Show description for 7d452ad
  

  gnecula

  authored and

  jax authors

  committed
  7d452ad

Commits on Dec 16, 2022

• (NFC) Prepare for migration from producing MHLO to producing StableHLO

  Show description for b8ae8e3
  

  Eugene Burmako

  authored and

  jax authors

  committed
  b8ae8e3

Commits on Dec 14, 2022

• Port type inference for 6 ops from StableHLO to MHLO

  Show description for ee1ad39
  

  Eugene Burmako

  authored and

  jax authors

  committed
  ee1ad39

• Raise error for unsupported shape polymorphism for custom call and fallback lowering

  gnecula

  committed
  ac77405

Commits on Nov 30, 2022

• Remove an unnecessary include/ from pybind11 include paths.

  Show description for e835739
  

  hawkinsp

  authored and

  jax authors

  committed
  e835739

Commits on Nov 29, 2022

• Drop support for Python 3.7

  jakevdp

  committed
  cb62a31

• Run GPU presubmits via bazel test on the RBE cluster. This speeds up the build + testing significantly (upto 10x).

  Show description for 3e5a505
  

  yashk2810

  authored and

  jax authors

  committed
  3e5a505

Commits on Nov 28, 2022

• Fix cudnn_header OSS BUILD dep.

  Show description for c54bc90
  

  zhangqiaorjc

  authored and

  jax authors

  committed
  c54bc90

• Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module.

  Show description for 4d1c4bc
  

  zhangqiaorjc

  authored and

  jax authors

  committed
  4d1c4bc

Commits on Nov 23, 2022

• Use the remote_gpu tag which is inserted by TF's workspace2 when REMOTE_GPU_TESTING=1

  Show description for a4e8df7
  

  yashk2810

  authored and

  jax authors

  committed
  a4e8df7

• Set tf_exec_properties on OSS tests to use TF's gpu pool in the RBE cluster.

  Show description for 8e27057
  

  yashk2810

  authored and

  jax authors

  committed
  8e27057

• Rollback of "Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module."

  Show description for d1fbdbc
  

  jax authors

  committed
  d1fbdbc

• Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module.

  Show description for 78963b6
  

  zhangqiaorjc

  authored and

  jax authors

  committed
  78963b6

Commits on Nov 17, 2022

• Remove unused potrf kernels.

  Show description for 0324cac
  

  pschuh

  authored and

  jax authors

  committed
  0324cac

Commits on Nov 16, 2022

• jaxlib: fix mlir_hlo build rule

  jakevdp

  committed
  e7f4fe0

Commits on Nov 15, 2022

• Remove custom potrf kernels in favor of native XLA cholesky support.

  Show description for 7635df8
  

  pschuh

  authored and

  jax authors

  committed
  7635df8

Commits on Nov 11, 2022

• Sytrd solver and SytrdDescriptor should NOT be CUDA only

  

  Rahul Batra

  committed
  31d8f62

Commits on Nov 10, 2022

• Add a GPU implementation of symmetric (Hermitian) tridiagonal reduction.

  Show description for 352b042
  

  hawkinsp

  authored and

  jax authors

  committed
  352b042

Commits on Nov 9, 2022

• Add support for Hessenberg and tridiagonal matrix reductions on CPU.

  Show description for 1cead77
  

  hawkinsp

  authored and

  jax authors

  committed
  1cead77

Commits on Nov 4, 2022

• [sparse] Update the guard of cusparse SpMM and SpMv algorithms to cusparse version 11.7.1 onwards.

  Show description for 46368e4
  

  tlu7

  authored and

  jax authors

  committed
  46368e4

Commits on Nov 1, 2022

• [sparse] Update the default CUSPARSE SPMV and SPMM algorithms in jaxlib.

  Show description for ef0f64e
  

  tlu7

  authored and

  jax authors

  committed
  ef0f64e

Commits on Oct 31, 2022

• Rollback of:

  Show description for 06c1d8e
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  06c1d8e

Commits on Oct 27, 2022

• [sparse] Update the default CUSPARSE SPMV and SPMM algorithms in jaxlib.

  Show description for 66e75ed
  

  tlu7

  authored and

  jax authors

  committed
  66e75ed

Commits on Oct 26, 2022

• Fix FP8 compilation failure in jaxlib stemming from the CUDA/ROCM merge.

  Show description for 0814770
  

  hawkinsp

  authored and

  jax authors

  committed
  0814770

Commits on Oct 25, 2022

• Merge CUDA and ROCM kernel code in jaxlib.

  Show description for a852710
  

  hawkinsp

  authored and

  jax authors

  committed
  a852710

Commits on Oct 24, 2022

• Move jaxlib cpu kernels under jaxlib/cpu/.

  Show description for 894093c
  

  hawkinsp

  authored and

  jax authors

  committed
  894093c

Commits on Oct 21, 2022

• Roll-back jax-ml#12892 because CUSPARSE_SPMV_COO_ALG2 is not available in CUDA 11.1

  Show description for e219d55
  

  tlu7

  authored and

  jax authors

  committed
  e219d55

Commits on Oct 20, 2022

• [sparse] Update the default cuSparse matvec algorithm in jaxlib.

  Show description for 7093142
  

  tlu7

  authored and

  jax authors

  committed
  7093142

Pagination

Previous

Next