Relax cuda XLA_TARGET
jonatanklosko opened this issue · comments
Instead of cuda118
and cuda120
we can just do cuda11
and cuda12
, which is what jaxlib does now. Then we just list what are the minimal supported versions for each.
Pre-compiled XLA extension
jonatanklosko opened this issue · comments
Instead of cuda118
and cuda120
we can just do cuda11
and cuda12
, which is what jaxlib does now. Then we just list what are the minimal supported versions for each.