diff --git a/ggml/src/ggml-cuda/ggml-cuda.cu b/ggml/src/ggml-cuda/ggml-cuda.cu index 676fa7929f..8b9330d633 100644 --- a/ggml/src/ggml-cuda/ggml-cuda.cu +++ b/ggml/src/ggml-cuda/ggml-cuda.cu @@ -4999,7 +4999,7 @@ static bool ggml_backend_cuda_device_supports_op(ggml_backend_dev_t dev, const g #ifdef GGML_USE_MUSA return false; #else - // KDA is faster using the AR kernel even when n_tokens >= 512. + // KDA is faster using the AR kernel even when n_tokens >= 512 //TODO: Add chunked kernel return op->src[0]->ne[2] == 1 || op->src[3]->ne[0] == op->src[2]->ne[0]; #endif // GGML_USE_MUSA