mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 04:16:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/rocm
History
Peter Hawkins 5617a02fa4 Remove JAX custom call implementation of batched triangular solve. 
XLA supports batched triangular solve on GPU and has since February 2022, which is older than the minimum jaxlib version. We can therefore delete our implementation and just use XLA's implementation.

PiperOrigin-RevId: 482031830
2022-10-18 15:04:14 -07:00
..
BUILD.bazel
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_gpu_kernel_helpers.cc
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_gpu_kernel_helpers.h
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_linalg.cc
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_lu_pivot_kernels.cc
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_lu_pivot_kernels.h
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_lu_pivot_kernels.hip.cc
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_prng_kernels.cc
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_prng_kernels.h
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_prng_kernels.hip.cc
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hip_prng.cc
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hipblas_kernels.cc
Remove JAX custom call implementation of batched triangular solve.
2022-10-18 15:04:14 -07:00
hipblas_kernels.h
Remove JAX custom call implementation of batched triangular solve.
2022-10-18 15:04:14 -07:00
hipblas.cc
Remove JAX custom call implementation of batched triangular solve.
2022-10-18 15:04:14 -07:00
hipsolver_kernels.cc
Revert: Use input-output aliasing for jaxlib GPU custom calls.
2022-10-07 14:36:24 -07:00
hipsolver_kernels.h
[ROCm] Upgrade to ROCm 5.3 and associated enhancements
2022-10-01 04:45:26 -07:00
hipsolver.cc
[ROCm] Upgrade to ROCm 5.3 and associated enhancements
2022-10-01 04:45:26 -07:00
hipsparse_kernels.cc
Revert: Use input-output aliasing for jaxlib GPU custom calls.
2022-10-07 14:36:24 -07:00
hipsparse_kernels.h
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
hipsparse.cc
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00