mirror of
https://github.com/leejet/stable-diffusion.cpp.git
synced 2026-11-12 11:56:36 +00:00
refactor: register RoPE kernels and fusion outside ggml
This commit is contained in:
1 parent
02e664d12d
commit
f4a1902061
7 files changed
+336
-5
No files matched your search
@@ -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
|
||||
Reference in new issue
Block a user