mirror of
https://github.com/microsoft/BitNet.git
synced 2026-05-03 11:20:36 +00:00
potassiummmm
50 lines
2.0 KiB
C
50 lines
2.0 KiB
C
#pragma once
|
|
|
|
#include "ggml.h"
|
|
#include "ggml-backend.h"
|
|
|
|
#ifdef __ARM_NEON
|
|
#include <arm_neon.h>
|
|
typedef float32_t bitnet_float_type;
|
|
#else
|
|
typedef float bitnet_float_type;
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
struct bitnet_tensor_extra {
|
|
int lut_scales_size;
|
|
int BK;
|
|
int n_tile_num;
|
|
uint8_t * qweights;
|
|
bitnet_float_type * scales;
|
|
};
|
|
|
|
GGML_API void ggml_bitnet_init(void);
|
|
GGML_API void ggml_bitnet_free(void);
|
|
// src0->type == Q4_0/IQ2_XXS/IQ3_XXS
|
|
// bitnet.cpp currently only supports BitNet quantization or GPTQ-like quantization (only scales, without zeros)
|
|
// If use i-quantization gguf models, the results will be wrong
|
|
// TODO: add customized block types Q2_0/Q3_0
|
|
GGML_API bool ggml_bitnet_can_mul_mat(const struct ggml_tensor * src0, const struct ggml_tensor * src1, const struct ggml_tensor * dst);
|
|
GGML_API size_t ggml_bitnet_mul_mat_get_wsize(const struct ggml_tensor * src0, const struct ggml_tensor * src1, const struct ggml_tensor * dst);
|
|
GGML_API void ggml_bitnet_mul_mat_task_init(void * src1, void * qlut, void * lut_scales, void * lut_biases, int n, int k, int m, int bits);
|
|
GGML_API void ggml_bitnet_mul_mat_task_compute(void * src0, void * scales, void * qlut, void * lut_scales, void * lut_biases, void * dst, int n, int k, int m, int bits);
|
|
GGML_API void ggml_bitnet_transform_tensor(struct ggml_tensor * tensor);
|
|
GGML_API int ggml_bitnet_get_type_bits(enum ggml_type type);
|
|
GGML_API void ggml_bitnet_set_n_threads(int n_threads);
|
|
#if defined(GGML_BITNET_ARM_TL1)
|
|
GGML_API void ggml_qgemm_lut(int m, int k, void* A, void* LUT, void* Scales, void* LUT_Scales, void* C);
|
|
GGML_API void ggml_preprocessor(int m, int k, void* B, void* LUT_Scales, void* QLUT);
|
|
#endif
|
|
#if defined(GGML_BITNET_X86_TL2)
|
|
GGML_API void ggml_qgemm_lut(int bs, int m, int k, int BK, void* A, void* sign, void* LUT, void* Scales, void* LUT_Scales, void* C);
|
|
GGML_API void ggml_preprocessor(int bs, int m, int three_k, int two_k, void* B, void* LUT_Scales, void* Three_QLUT, void* Two_QLUT);
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|