refactor: register RoPE kernels and fusion outside ggml

This commit is contained in:
leejet committed 2026-10-31 18:18:33 +08:00
1 parent 02e664d12d
commit f4a1902061
7 files changed
+336 -5

No files matched your search

+12
View File
@@ -0,0 +1,12 @@
#ifndef __SD_CORE_GGML_EXTENSIONS_H__
#define __SD_CORE_GGML_EXTENSIONS_H__
#include "ggml-backend.h"
void sd_ggml_register_extensions(ggml_backend_t backend);
#ifndef SD_USE_UPSTREAM_GGML
ggml_tensor* sd_ggml_rope_apply(ggml_context* ctx, ggml_backend_t backend, ggml_tensor* x, ggml_tensor* pe, bool interleaved);
#endif
#endif