upstream-archive/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-10-02 19:07:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
leejet
2026-09-19 17:50:01 +08:00
committed by GitHub
parent 2ea8aff7ef
commit 3e037a81e4
9 changed files with 24 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/model/common/ggml_block.hpp
+1 -1
View File
@@ -728,7 +728,7 @@ public:
std::get<2>(stride), std::get<1>(stride), std::get<0>(stride),
std::get<2>(padding), std::get<1>(padding), std::get<0>(padding),
std::get<2>(dilation), std::get<1>(dilation), std::get<0>(dilation),
force_prec_f32);
force_prec_f32, ctx->conv3d_direct_enabled);
}
};
src/model/vae/wan_vae.hpp
+2 -1
View File
@@ -78,7 +78,8 @@ namespace WAN {
return ggml_ext_conv_3d(ctx->ggml_ctx, ctx->backend, x, w, b, in_channels,
std::get<2>(stride), std::get<1>(stride), std::get<0>(stride),
0, 0, 0,
std::get<2>(dilation), std::get<1>(dilation), std::get<0>(dilation));
std::get<2>(dilation), std::get<1>(dilation), std::get<0>(dilation),
false, ctx->conv3d_direct_enabled);
}
};