mirror of
https://github.com/likelovewant/ollama-for-amd.git
synced 2025-12-21 14:26:30 +00:00
49a9c9ba6a
* Revert "add support for NVIDIA Nemotron 3 Nano" This reverts commit e7d2ae9d69421012e9a8765c06a3fdf0e45b12f3. * GGML update to 380b4c984 Remove MaskBatchPadding as GGML_KQ_MASK_PAD is no longer present (no padding required) * update to c45f89d55 * ec98e2002 solar pro needed more adjusting - needs verification * review comments
26 lines
1.1 KiB
Diff
26 lines
1.1 KiB
Diff
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
|
|
From: Michael Yang <git@mxy.ng>
|
|
Date: Tue, 18 Nov 2025 11:13:04 -0800
|
|
Subject: [PATCH] ggml-cuda: skip large batches
|
|
|
|
cuda panics on batches larger than 1024 so mark it as unsupported to
|
|
fallback to cpu
|
|
---
|
|
ggml/src/ggml-cuda/ggml-cuda.cu | 3 +++
|
|
1 file changed, 3 insertions(+)
|
|
|
|
diff --git a/ggml/src/ggml-cuda/ggml-cuda.cu b/ggml/src/ggml-cuda/ggml-cuda.cu
|
|
index 48cdb1dcf..3102d7ea7 100644
|
|
--- a/ggml/src/ggml-cuda/ggml-cuda.cu
|
|
+++ b/ggml/src/ggml-cuda/ggml-cuda.cu
|
|
@@ -4633,6 +4633,9 @@ static bool ggml_backend_cuda_device_supports_op(ggml_backend_dev_t dev, const g
|
|
if (b->type == GGML_TYPE_F16 && a->type != GGML_TYPE_F16) {
|
|
return false;
|
|
}
|
|
+ if (op->op == GGML_OP_MUL_MAT && b->ne[2] * b->ne[3] > 1024) {
|
|
+ return false;
|
|
+ }
|
|
#ifdef GGML_USE_MUSA
|
|
const int cc = ggml_cuda_info().devices[dev_ctx->device].cc;
|
|
if (b->ne[2]*b->ne[3] > 1 && !ggml_is_transposed(a) && !ggml_is_transposed(b)) {
|