mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-26 20:16:05 +00:00
Code Issues Packages Projects Releases Wiki Activity
25,661 Commits 508 Branches 263 Tags
194884d311
Commit Graph

2 Commits

Author SHA1 Message Date
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
Paweł Paruzel
23fdb91252 Port Schur Decomposition to XLA's FFI 
This CL only contains the C++ changes. Python lowering code will be added after the forward compatibility window of 3 weeks.

PiperOrigin-RevId: 685689593
 2024-10-14 06:46:42 -07:00