From 981e5ea0d7579b4803c86afbb09a7cd7d7bf3bb8 Mon Sep 17 00:00:00 2001
From: Samuel Oliveira Alves
 <107287165+SamuelOliveirads@users.noreply.github.com>
Date: Thu, 13 Aug 2026 13:02:46 -0300
Subject: [PATCH] DSpark: gather BF16 Markov rows (#2304)

* DSpark: gather BF16 Markov rows

* DSpark: gather Markov rows by type
---
 src/graphs/build_dflash.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/graphs/build_dflash.cpp b/src/graphs/build_dflash.cpp
index 821b5c03..87030f65 100644
--- a/src/graphs/build_dflash.cpp
+++ b/src/graphs/build_dflash.cpp
@@ -26,7 +26,8 @@ ggml_tensor * llm_build_context::build_dspark_logits(
     ggml_tensor * draft_tokens = nullptr;
 
     for (int64_t i = 0; i < n_tokens; ++i) {
-        ggml_tensor * markov_w1 = ggml_get_rows(ctx0, model.dspark_markov_w1, previous);
+        ggml_tensor * markov_w1 = ggml_get_rows_ext(
+                ctx0, model.dspark_markov_w1, previous, true, false);
         ggml_tensor * markov_bias = ggml_mul_mat(ctx0, model.dspark_markov_w2, markov_w1);
         ggml_tensor * base_row = ggml_view_2d(
                 ctx0,