Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
In my development setup, I have different machines with different GPUs. They share the home directory on a network filesystem. When I switch between machines, since the JIT compilation flags change, I'll have to recompile kernels every time. One solution is to specify the same `TORCH_CUDA_ARCH_LIST` every time. However, I keep forgetting that. Another solution, as proposed in this PR, is to put different arch list in different cache directory.
- Loading branch information