DarthAffe/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-09-24 20:20:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: mark graph cuts with both a prefix and a suffix (#1883)

Browse Source
This commit is contained in:
Wagner Bruna 2026-08-19 11:27:48 -03:00 committed by GitHub
parent 1706b32813
commit 88b044be7f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/core/ggml_graph_cut.cpp
View File

@ -453,11 +453,12 @@ namespace sd::ggml_graph_cut {
if (tensor == nullptr || tensor->name[0] == '\0') { if (tensor == nullptr || tensor->name[0] == '\0') {
return false; return false;
} }
return std::strncmp(tensor->name, GGML_RUNNER_CUT_PREFIX, std::strlen(GGML_RUNNER_CUT_PREFIX)) == 0; return starts_with(tensor->name, GGML_RUNNER_CUT_PREFIX) &&
ends_with(tensor->name, GGML_RUNNER_CUT_SUFFIX);
} }
std::string make_graph_cut_name(const std::string& group, const std::string& output) { std::string make_graph_cut_name(const std::string& group, const std::string& output) {
return std::string(GGML_RUNNER_CUT_PREFIX) + group + "|" + output; return std::string(GGML_RUNNER_CUT_PREFIX) + group + "|" + output + GGML_RUNNER_CUT_SUFFIX;
} }
void mark_graph_cut(ggml_tensor* tensor, const std::string& group, const std::string& output) { void mark_graph_cut(ggml_tensor* tensor, const std::string& group, const std::string& output) {
@ -783,7 +784,9 @@ namespace sd::ggml_graph_cut {
plan.has_cuts = true; plan.has_cuts = true;
std::string full_name(node->name); std::string full_name(node->name);
std::string payload = full_name.substr(std::strlen(GGML_RUNNER_CUT_PREFIX)); size_t prefix_len = std::strlen(GGML_RUNNER_CUT_PREFIX);
size_t suffix_len = std::strlen(GGML_RUNNER_CUT_SUFFIX);
std::string payload = full_name.substr(prefix_len, full_name.size() - prefix_len - suffix_len);
size_t sep = payload.find('|'); size_t sep = payload.find('|');
std::string group = sep == std::string::npos ? payload : payload.substr(0, sep); std::string group = sep == std::string::npos ? payload : payload.substr(0, sep);

1
src/core/ggml_graph_cut.h
View File

@ -68,6 +68,7 @@ namespace sd::ggml_graph_cut {
}; };
static constexpr const char* GGML_RUNNER_CUT_PREFIX = "ggml_runner_cut:"; static constexpr const char* GGML_RUNNER_CUT_PREFIX = "ggml_runner_cut:";
static constexpr const char* GGML_RUNNER_CUT_SUFFIX = "|";
struct MaxVramAssignment { struct MaxVramAssignment {
float default_gib = 0.f; float default_gib = 0.f;