CUDA: fuse SSM_CONV + ADD(bias) + SILU (#22478)

This commit is contained in:
Anav Prasad
2026-04-30 02:39:56 +08:00
committed by GitHub
parent 683c5acb90
commit 098705a29e
4 changed files with 129 additions and 8 deletions
+1 -1
View File
@@ -1,3 +1,3 @@
#include "common.cuh"
void ggml_cuda_op_ssm_conv(ggml_backend_cuda_context & ctx, ggml_tensor * dst, ggml_tensor * silu_dst = nullptr);
void ggml_cuda_op_ssm_conv(ggml_backend_cuda_context & ctx, ggml_tensor * dst, ggml_tensor * bias_add_node = nullptr, ggml_tensor * silu_dst = nullptr);