ggml : fix conv 2d dw (#25490)

2026-07-09 17:56:32 +03:00
parent c15c5c77a4
commit 5c3a586860
2 changed files with 6 additions and 3 deletions
@@ -4917,7 +4917,9 @@ static bool ggml_backend_cuda_device_supports_op(ggml_backend_dev_t dev, const g
        case GGML_OP_IM2COL:
        case GGML_OP_IM2COL_3D:
        case GGML_OP_CONV_2D:
+            return true;
        case GGML_OP_CONV_2D_DW:
+            return op->src[0]->type == GGML_TYPE_F32;
        case GGML_OP_CONV_TRANSPOSE_2D:
        case GGML_OP_POOL_2D:
            return true;