Skip to content

Commit 6e522ec

Browse files
committed
fix compile
1 parent f44c177 commit 6e522ec

File tree

4 files changed

+8
-3
lines changed

4 files changed

+8
-3
lines changed

ggml/include/ggml.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -517,6 +517,7 @@ extern "C" {
517517
GGML_OP_CROSS_ENTROPY_LOSS,
518518
GGML_OP_CROSS_ENTROPY_LOSS_BACK,
519519
GGML_OP_OPT_STEP_ADAMW,
520+
520521
GGML_OP_FILL,
521522

522523
GGML_OP_COUNT,

ggml/src/ggml-cpu/ops.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6848,7 +6848,6 @@ static void ggml_compute_forward_fill_f32(
68486848
const int nc = dst->ne[0];
68496849

68506850
const size_t nb00 = dst->nb[0];
6851-
const size_t nb01 = dst->nb[1];
68526851

68536852
const size_t nb0 = dst->nb[0];
68546853
const size_t nb1 = dst->nb[1];

ggml/src/ggml-cpu/ops.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -73,6 +73,7 @@ void ggml_compute_forward_upscale(const struct ggml_compute_params * params, str
7373
void ggml_compute_forward_pad(const struct ggml_compute_params * params, struct ggml_tensor * dst);
7474
void ggml_compute_forward_pad_reflect_1d(const struct ggml_compute_params * params, struct ggml_tensor * dst);
7575
void ggml_compute_forward_arange(const struct ggml_compute_params * params, struct ggml_tensor * dst);
76+
void ggml_compute_forward_fill(const struct ggml_compute_params * params, struct ggml_tensor * dst);
7677
void ggml_compute_forward_timestep_embedding(const struct ggml_compute_params * params, struct ggml_tensor * dst);
7778
void ggml_compute_forward_argsort(const struct ggml_compute_params * params, struct ggml_tensor * dst);
7879
void ggml_compute_forward_leaky_relu(const struct ggml_compute_params * params, struct ggml_tensor * dst);

ggml/src/ggml.c

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -982,9 +982,11 @@ static const char * GGML_OP_NAME[GGML_OP_COUNT] = {
982982
"CROSS_ENTROPY_LOSS",
983983
"CROSS_ENTROPY_LOSS_BACK",
984984
"OPT_STEP_ADAMW",
985+
986+
"FILL",
985987
};
986988

987-
static_assert(GGML_OP_COUNT == 82, "GGML_OP_COUNT != 82");
989+
static_assert(GGML_OP_COUNT == 83, "GGML_OP_COUNT != 83");
988990

989991
static const char * GGML_OP_SYMBOL[GGML_OP_COUNT] = {
990992
"none",
@@ -1077,9 +1079,11 @@ static const char * GGML_OP_SYMBOL[GGML_OP_COUNT] = {
10771079
"cross_entropy_loss(x,y)",
10781080
"cross_entropy_loss_back(x,y)",
10791081
"adamw(x)",
1082+
1083+
"fill(x)",
10801084
};
10811085

1082-
static_assert(GGML_OP_COUNT == 82, "GGML_OP_COUNT != 82");
1086+
static_assert(GGML_OP_COUNT == 83, "GGML_OP_COUNT != 83");
10831087

10841088
static_assert(GGML_OP_POOL_COUNT == 2, "GGML_OP_POOL_COUNT != 2");
10851089

0 commit comments

Comments
 (0)