cuda/vulkan: specify fp32-only support for some operations in supports_op (ggml/1129)
ggml-ci
This commit is contained in:
parent
8371d44595
commit
0cbee131ad
3 changed files with 8 additions and 10 deletions
|
@ -3155,7 +3155,7 @@ static bool ggml_backend_cuda_device_supports_op(ggml_backend_dev_t dev, const g
|
|||
return false;
|
||||
} break;
|
||||
case GGML_OP_SILU_BACK:
|
||||
return ggml_is_contiguous(op->src[0]);
|
||||
return ggml_is_contiguous(op->src[0]) && op->src[0]->type == GGML_TYPE_F32;
|
||||
break;
|
||||
case GGML_OP_NORM:
|
||||
case GGML_OP_RMS_NORM:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue