chenchenghao/DeepGEMM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add sm_100f support and make nvcc 13 happy (#157)

Browse Source 
Signed-off-by: Xiwen Yu <13230610+VALLIS-NERIA@users.noreply.github.com>
This commit is contained in:
xiweny
2025-08-22 17:19:32 +08:00
committed by GitHub
parent f85ec649d7
commit affdb1cd90
4 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
csrc/jit/compiler.hpp
View File

@@ -155,7 +155,7 @@ public:
signature = fmt::format("NVCC{}.{}", nvcc_major, nvcc_minor);
// The override the compiler flags
flags = fmt::format("{} -I{} --gpu-architecture=sm_{}a "
flags = fmt::format("{} -I{} --gpu-architecture=sm_{} "
"--compiler-options=-fPIC,-O3,-fconcepts,-Wno-deprecated-declarations,-Wno-abi "
"-cubin -O3 --expt-relaxed-constexpr --expt-extended-lambda",
flags, library_include_path.c_str(), device_runtime->get_arch());
@@ -205,7 +205,7 @@ public:
}
// Override the compiler flags
flags = fmt::format("{} {}--gpu-architecture=sm_{}a -default-device {}",
flags = fmt::format("{} {}--gpu-architecture=sm_{} -default-device {}",
flags, include_dirs, device_runtime->get_arch(), pch_flags);
}

7
csrc/jit/device_runtime.hpp
View File

@@ -25,9 +25,12 @@ public:
return {prop->major, prop->minor};
}
int get_arch() {
std::string get_arch() {
const auto& [major, minor] = get_arch_pair();
return major * 10 + minor;
if (major == 10 && minor != 1) {
return "100f";
}
return std::to_string(major * 10 + minor) + "a";
}
int get_arch_major() {