Jon Kyl issues

Repositories
Issues
Comments

Results 1 issues of


                                            Jon Kyl

Device -> host transfer causes CUDA OOM on WSL2

### Description #### Summary: On WSL2, after running inference with a loaded checkpoint, JAX array transfers from device to host fail with `CUDA_ERROR_OUT_OF_MEMORY`, despite: - Tuning XLA flags like preallocate...

bug