fix cann compile error
This commit is contained in:
parent
c8d89317c9
commit
265cb43538
|
|
@ -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
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue