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

Code lint

Browse Source
This commit is contained in:
Chenggang Zhao
2025-07-21 11:00:50 +08:00
parent 436a56314c
commit 187656694f
2 changed files with 21 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
csrc/jit/compiler.hpp
+1 -18
View File
@@ -19,26 +19,9 @@ class Compiler {
std::string library_version;
std::filesystem::path library_root_path;
static void collect_files(const std::filesystem::path& dir,
std::vector<std::filesystem::path>& out) {
// We met ABI breakage using std::filesystem::recursive_directory_iterator
// Use std::filesystem::directory_iterator instead
for (const auto& entry : std::filesystem::directory_iterator(dir)) {
if (entry.is_directory()) {
collect_files(entry.path(), out);
} else if (entry.is_regular_file() && entry.path().extension()==".cuh") {
out.emplace_back(entry.path());
}
}
}
std::string get_library_version() const {
std::vector<std::filesystem::path> files;
collect_files(library_include_path / "deep_gemm", files);
std::sort(files.begin(), files.end());
std::stringstream ss;
for (const auto& f : files) {
for (const auto& f: collect_files(library_include_path / "deep_gemm")) {
std::ifstream in(f, std::ios::binary);
ss << in.rdbuf();
}