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

fix: layer split across cuda (#2262)

Browse Source
This commit is contained in:
cumal authored and GitHub committed 2026-11-03 00:03:43 +08:00
1 parent 1350cf5097
commit 250ea6e385
1 file changed
+10 -10
Download Patch File Download Diff File Expand all files Collapse all files
src/core/layer_split_partition.cpp
+10 -10
View File
@@ -206,23 +206,23 @@ namespace sd {
for (size_t seg_idx = 0; seg_idx < plan.segments.size(); seg_idx++) {
ggml_backend_t backend = backend_by_segment[seg_idx];
const auto& segment = plan.segments[seg_idx];
for (int node_index : segment.internal_node_indices) {
auto assign_node = [&](int node_index) {
if (node_index < 0 || node_index >= n_nodes) {
continue;
return;
}
ggml_tensor* node = ggml_graph_node(gf, node_index);
if (node != nullptr) {
if (node == nullptr) {
return;
}
if (assignment.node_assignments.find(node) == assignment.node_assignments.end()) {
assignment.node_assignments[node] = backend;
}
};
for (int node_index : segment.internal_node_indices) {
assign_node(node_index);
}
for (int node_index : segment.output_node_indices) {
if (node_index < 0 || node_index >= n_nodes) {
continue;
}
ggml_tensor* node = ggml_graph_node(gf, node_index);
if (node != nullptr) {
assignment.node_assignments[node] = backend;
}
assign_node(node_index);
}
}