mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2025-11-14 11:07:10 +00:00
fix cann compile error
This commit is contained in:
@@ -2210,7 +2210,8 @@ static bool ggml_backend_cann_supports_op(ggml_backend_dev_t dev,
|
|||||||
case GGML_OP_COUNT_EQUAL:
|
case GGML_OP_COUNT_EQUAL:
|
||||||
return true;
|
return true;
|
||||||
case GGML_OP_SCALE:
|
case GGML_OP_SCALE:
|
||||||
float bias = ((const float *)(op->op_params))[1];
|
float bias;
|
||||||
|
memcpy(&bias, (float*)op->op_params + 1, sizeof(float));
|
||||||
return bias == 0.0f; // TODO: support bias != 0.0f
|
return bias == 0.0f; // TODO: support bias != 0.0f
|
||||||
case GGML_OP_SOFT_MAX:
|
case GGML_OP_SOFT_MAX:
|
||||||
// TODO: support broadcast
|
// TODO: support broadcast
|
||||||
|
|||||||
Reference in New Issue
Block a user