Skip to content

Commit

Permalink
f32 sigmoid in vulkan supports op
Browse files Browse the repository at this point in the history
  • Loading branch information
cmdr2 committed Feb 28, 2025
1 parent 3af766e commit c6f04b3
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/ggml-vulkan/ggml-vulkan.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -8371,6 +8371,7 @@ static bool ggml_backend_vk_device_supports_op(ggml_backend_dev_t dev, const ggm
case GGML_UNARY_OP_SILU:
case GGML_UNARY_OP_RELU:
case GGML_UNARY_OP_TANH:
case GGML_UNARY_OP_SIGMOID:
return ggml_is_contiguous(op->src[0]) && op->src[0]->type == GGML_TYPE_F32;
default:
return false;
Expand Down

0 comments on commit c6f04b3

Please sign in to comment.