ggml : build fixes

ggml-ci
This commit is contained in:
Georgi Gerganov 2023-12-07 13:52:27 +02:00
parent 95e9d8a780
commit 213a4e2c32
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
2 changed files with 2 additions and 3 deletions

View File

@ -797,7 +797,6 @@ static bool ggml_metal_supports_op(const struct ggml_tensor * op) {
default: default:
return false; return false;
} }
break;
case GGML_OP_NONE: case GGML_OP_NONE:
case GGML_OP_RESHAPE: case GGML_OP_RESHAPE:
case GGML_OP_VIEW: case GGML_OP_VIEW:
@ -827,7 +826,7 @@ static bool ggml_metal_supports_op(const struct ggml_tensor * op) {
case GGML_OP_GET_ROWS: case GGML_OP_GET_ROWS:
{ {
return op->ne[0] % 4 == 0; return op->ne[0] % 4 == 0;
} break; }
default: default:
return false; return false;
} }

2
ggml.c
View File

@ -14100,7 +14100,7 @@ static void ggml_compute_forward(struct ggml_compute_params * params, struct ggm
} break; } break;
case GGML_OP_SOFT_MAX: case GGML_OP_SOFT_MAX:
{ {
ggml_compute_forward_soft_max(params, tensor->src[0], tensor); ggml_compute_forward_soft_max(params, tensor->src[0], tensor->src[1], tensor);
} break; } break;
case GGML_OP_SOFT_MAX_BACK: case GGML_OP_SOFT_MAX_BACK:
{ {