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

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

No files matched your search

+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);
}
};