(nvcc_extra_args)
| 53 | |
| 54 | |
| 55 | def append_nvcc_threads(nvcc_extra_args): |
| 56 | _, bare_metal_version = get_cuda_bare_metal_version(CUDA_HOME) |
| 57 | if bare_metal_version >= Version("11.2"): |
| 58 | return nvcc_extra_args + ["--threads", "4"] |
| 59 | return nvcc_extra_args |
| 60 | |
| 61 | |
| 62 | if not torch.cuda.is_available(): |
no test coverage detected