[DRIVER] Fixed CUDA 10.1 bug (#357)

This commit is contained in:
Philippe Tillet
2021-10-26 11:17:06 -07:00
committed by GitHub
parent 5ce1b726dc
commit d35014ba47

View File

@@ -92,7 +92,7 @@ int vptx(int version){
if(version >= 11010) return 71; if(version >= 11010) return 71;
if(version >= 11000) return 70; if(version >= 11000) return 70;
if(version >= 10020) return 65; if(version >= 10020) return 65;
if(version >= 10010) return 64; if(version >= 10010) return 63;
if(version >= 10000) return 63; if(version >= 10000) return 63;
throw std::runtime_error("Triton requires CUDA 10+"); throw std::runtime_error("Triton requires CUDA 10+");
} }