vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Address comments

Browse source
This commit is contained in:
Haohui Mai 2023-11-20 22:04:21 -08:00
parent c6874f9759
commit 566785f560
2 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

2
ggml-cuda.cu
View file

@ -8057,7 +8057,7 @@ bool ggml_cuda_compute_forward(struct ggml_compute_params * params, struct ggml_
if (tensor->op == GGML_OP_MUL_MAT) {
if (tensor->src[0]->ne[3] != tensor->src[1]->ne[3]) {
#ifndef NDEBUG
fprintf(stderr, "%s: cannot compute %s: src0->ne[3] = %ld, src1->ne[3] = %ld - fallback to CPU\n", __func__, tensor->name, tensor->src[0]->ne[3], tensor->src[1]->ne[3]);
fprintf(stderr, "%s: cannot compute %s: src0->ne[3] = " PRId64 ", src1->ne[3] = " PRId64 " - fallback to CPU\n", __func__, tensor->name, tensor->src[0]->ne[3], tensor->src[1]->ne[3]);
#endif
return false;
}

4
ggml.c
View file

@ -1586,7 +1586,7 @@ inline static void ggml_vec_argmax_f32(const int n, int * s, const float * x) {
// data types
//
static const char * GGML_OP_NAME[GGML_OP_COUNT + 2] = {
static const char * GGML_OP_NAME[GGML_OP_COUNT] = {
"NONE",
"DUP",
@ -1664,8 +1664,6 @@ static const char * GGML_OP_NAME[GGML_OP_COUNT + 2] = {
"CROSS_ENTROPY_LOSS",
"CROSS_ENTROPY_LOSS_BACK",
"",
"",
};
static_assert(GGML_OP_COUNT == 68, "GGML_OP_COUNT != 68");