Get cuda version of nvcc.
()
| 142 | |
| 143 | |
| 144 | def get_nvcc_version(): |
| 145 | """ |
| 146 | Get cuda version of nvcc. |
| 147 | """ |
| 148 | nvcc_output = subprocess.check_output(["nvcc", "--version"], universal_newlines=True) |
| 149 | output = nvcc_output.split() |
| 150 | release_idx = output.index("release") + 1 |
| 151 | nvcc_cuda_version = float(output[release_idx].split(",")[0]) |
| 152 | return nvcc_cuda_version |
| 153 | |
| 154 | |
| 155 | def get_gencode_flags(archs): |
no test coverage detected