mirror of
https://github.com/ikawrakow/ik_llama.cpp.git
synced 2026-04-30 03:11:51 +00:00
34 lines
766 B
C
34 lines
766 B
C
//
|
|
// Copyright (C) 2025 Iwan Kawrakow
|
|
// MIT license
|
|
// SPDX-License-Identifier: MIT
|
|
//
|
|
|
|
#pragma once
|
|
#include <stdint.h>
|
|
#include <stdbool.h>
|
|
#include "iqk_config.h"
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
struct ggml_tensor;
|
|
|
|
void iqk_sumrows_div(struct ggml_tensor * div, int ith, int nth);
|
|
|
|
void iqk_grouped_top_k(struct ggml_tensor * dst, int ith, int nth);
|
|
|
|
void iqk_argsort(struct ggml_tensor * dst, int ith, int nth);
|
|
|
|
void iqk_bailingmoev2_experts(struct ggml_tensor * dst, struct ggml_tensor * topk, int ith, int nth);
|
|
|
|
void iqk_glm45moe_experts(struct ggml_tensor * dst, struct ggml_tensor * topk_view, int ith, int nth);
|
|
|
|
void iqk_openai_experts(struct ggml_tensor * topk, struct ggml_tensor * softmax, int ith, int nth);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
|