cuda : update supports_op for matrix multiplication (#8245)

2024-07-02 08:39:38 +02:00 · 2024-07-02 08:39:38 +02:00 · 0e0590adab
commit 0e0590adab
parent a9f3b10215
2 changed files with 31 additions and 17 deletions
--- a/tests/test-backend-ops.cpp
+++ b/tests/test-backend-ops.cpp
@ -2052,6 +2052,7 @@ static bool test_backend(ggml_backend_t backend, test_mode mode, const char * op
        GGML_TYPE_IQ2_XS, GGML_TYPE_IQ2_S,
        GGML_TYPE_IQ3_XXS, GGML_TYPE_IQ1_S, GGML_TYPE_IQ1_M,
        GGML_TYPE_IQ4_NL, GGML_TYPE_IQ3_S, GGML_TYPE_IQ4_XS,
+        GGML_TYPE_BF16,
    };

    // unary ops