mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/cpu
History
Sergei Lebedev 194884d311 Migrated to mypy 1.14.1 with --allow_redefinition 
I initially wanted to upgrade to 1.15, but it seems to have a bug in how
ternary expressions are type checked. For example,

   def f(x: int) -> str: ...
   def g(x: int) -> str: ...

   callback = f if ... else g  # has type object!
2025-02-13 15:38:28 +00:00
..
_lapack
Migrated to mypy 1.14.1 with --allow_redefinition
2025-02-13 15:38:28 +00:00
BUILD
Use thread-safe initialization of LAPACK kernels.
2025-01-13 02:51:38 -08:00
cpu_kernels.cc
Add LAPACK lowering for lax.linalg.tridiagonal_solve on CPU.
2025-01-10 08:56:46 -08:00
lapack_kernels_using_lapack.cc
Add LAPACK lowering for lax.linalg.tridiagonal_solve on CPU.
2025-01-10 08:56:46 -08:00
lapack_kernels.cc
Add LAPACK lowering for lax.linalg.tridiagonal_solve on CPU.
2025-01-10 08:56:46 -08:00
lapack_kernels.h
Add LAPACK lowering for lax.linalg.tridiagonal_solve on CPU.
2025-01-10 08:56:46 -08:00
lapack.cc
Use thread-safe initialization of LAPACK kernels.
2025-01-13 02:51:38 -08:00