finish bitnet i2 e2e

This commit is contained in:
Eddie-Wang 2024-06-08 12:44:13 +00:00
parent 2a01a7ce0d
commit 4e1ab50628
6 changed files with 16 additions and 637 deletions

7
ggml.h
View file

@ -507,8 +507,6 @@ extern "C" {
GGML_OP_CROSS_ENTROPY_LOSS,
GGML_OP_CROSS_ENTROPY_LOSS_BACK,
GGML_OP_BITLINEAR_QUANT,
GGML_OP_COUNT,
};
@ -996,11 +994,6 @@ extern "C" {
struct ggml_context * ctx,
struct ggml_tensor * a);
// for bitnet
GGML_API struct ggml_tensor * ggml_bitlinear_quant(
struct ggml_context * ctx,
struct ggml_tensor * a);
// argmax along rows
GGML_API struct ggml_tensor * ggml_argmax(
struct ggml_context * ctx,