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

refactor: register CUDA affine fusion extensions

Browse Source
This commit is contained in:
leejet committed 2026-10-31 22:59:06 +08:00
1 parent 7ad74e6490
commit 5c5535d01d
5 files changed
+205 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

src/core/ggml_extensions.cpp
+2
View File
@@ -10,6 +10,7 @@
#include <unordered_set>
#ifdef SD_CUDA_EXTENSIONS
void sd_ggml_register_cuda_affine(ggml_backend_dev_t device);
void sd_ggml_register_cuda_i8(ggml_backend_dev_t device);
void sd_ggml_register_cuda_convrot(ggml_backend_dev_t device);
void sd_ggml_register_cuda_rope(ggml_backend_dev_t device);
@@ -182,6 +183,7 @@ void sd_ggml_register_extensions(ggml_backend_t backend) {
}
if (cuda) {
#ifdef SD_CUDA_EXTENSIONS
sd_ggml_register_cuda_affine(device);
sd_ggml_register_cuda_i8(device);
sd_ggml_register_cuda_convrot(device);
sd_ggml_register_cuda_rope(device);