From c103f2f4ef8e5e90caef71131a2e1cba88a50a72 Mon Sep 17 00:00:00 2001 From: FirstTimeEZ <179362031+FirstTimeEZ@users.noreply.github.com> Date: Mon, 18 Nov 2024 21:35:01 +1300 Subject: [PATCH] GGML_OP_SOFT_MAX --- ggml/src/ggml-vulkan/ggml-vulkan.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/ggml/src/ggml-vulkan/ggml-vulkan.cpp b/ggml/src/ggml-vulkan/ggml-vulkan.cpp index 72bdd13e3..57cc7747d 100644 --- a/ggml/src/ggml-vulkan/ggml-vulkan.cpp +++ b/ggml/src/ggml-vulkan/ggml-vulkan.cpp @@ -3934,8 +3934,7 @@ static vk_pipeline ggml_vk_op_get_pipeline(ggml_backend_vk_context * ctx, const if (src1 == nullptr || src1->type == GGML_TYPE_F32) { return ctx->device->pipeline_soft_max_f32; } - else { - GGML_ASSERT(src1->type == GGML_TYPE_F16); + else if (src1->type == GGML_TYPE_F16) { return ctx->device->pipeline_soft_max_f32_f16; } }