Sudhakar Singh
Sudhakar Singh
Looking into it
@hawkinsp it does look like it's a resolved bug in CUDA11.2 https://docs.nvidia.com/cuda/archive/11.2.0/cuda-toolkit-release-notes/index.html#cufft-resolved-issues
Yes, that's right. Where should we document this? (in JAX README perhaps - somewhere in [jax:cuda installation steps](https://github.com/google/jax#pip-installation-gpu-cuda)?) Edit: Ah, I think you meant adding a warning in the code...
@PgLoLo was this resolved?
Is that straightforward (basically, are the upstream single host pjit tests extensible enough that they could be run on multi-host multi-GPU scenario)? I'm asking this because these tests needed to...
addressed the concerns
fixed lint and doc build issues now
Done with making changes as requested
ping after fixing nits
Was this issue resolved? @akhilnadigatla