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

perf: accelerate VAE direct 3D convolutions (#1996)

Browse Source
This commit is contained in:
leejet authored and GitHub committed 2026-09-19 17:50:01 +08:00
1 parent 2ea8aff7ef
commit 3e037a81e4
9 files changed
+24 -8
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/core/ggml_runner.h
+6
View File
@@ -71,6 +71,7 @@ struct GGMLRunnerContext {
float linear_scale = 0.f;
float attn_scale = 0.f;
bool conv2d_direct_enabled = false;
bool conv3d_direct_enabled = false;
bool circular_x_enabled = false;
bool circular_y_enabled = false;
ggml_tensor* ip_context = nullptr;
@@ -178,6 +179,7 @@ protected:
float linear_scale = 0.f;
float attn_scale = 0.f;
bool conv2d_direct_enabled = false;
bool conv3d_direct_enabled = false;
bool circular_x_enabled = false;
bool circular_y_enabled = false;
@@ -346,6 +348,10 @@ public:
conv2d_direct_enabled = enabled;
}
void set_conv3d_direct_enabled(bool enabled) {
conv3d_direct_enabled = enabled;
}
void set_circular_axes(bool circular_x, bool circular_y) {
circular_x_enabled = circular_x;
circular_y_enabled = circular_y;