Now that device is used in backend_init_func, it must always be set

jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton

Apache License 2.0

354 stars 40 forks source link

Closed copybara-service[bot] closed 3 months ago

copybara-service[bot] commented 3 months ago

Now that device is used in backend_init_func, it must always be set and not only if compute_capability is None.