Relax cuda XLA_TARGET - Githubissues

elixir-nx / xla

Pre-compiled XLA extension

Apache License 2.0

83 stars 21 forks source link

Relax cuda XLA_TARGET #55

Closed jonatanklosko closed 1 month ago

jonatanklosko commented 9 months ago

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.