fix unit test failure

chraac · chraac · commit 560729ed6fa0 · 2025-07-12T00:39:14.000+08:00
diff --git a/ggml/src/ggml-qnn/npu/device/op_flash_attn.cpp b/ggml/src/ggml-qnn/npu/device/op_flash_attn.cpp
@@ -105,7 +105,9 @@ void flash_attn_impl(hexagon::tensor * out, const hexagon::tensor * q, const hex
         }
 
         const npu_device_fp16_t * mp =
-            mask_ptr ? reinterpret_cast<const npu_device_fp16_t *>(mask_ptr + iq1 * mask->get_nb(1)) : nullptr;
+            mask_ptr ? reinterpret_cast<const npu_device_fp16_t *>(mask_ptr + iq1 * mask->get_nb(1) +
+                                                                   (iq3 % mask->get_ne(2)) * mask->get_nb(2)) :
+                       nullptr;
 
         // k indices
         const int ik3 = iq3 / rk3;

Original file line number	Diff line number	Diff line change
`@@ -105,7 +105,9 @@ void flash_attn_impl(hexagon::tensor * out, const hexagon::tensor * q, const hex`
`105`	`105`	`}`
`106`	`106`
`107`	`107`	`const npu_device_fp16_t * mp =`
`108`		`- mask_ptr ? reinterpret_cast<const npu_device_fp16_t >(mask_ptr + iq1 mask->get_nb(1)) : nullptr;`
	`108`	`+ mask_ptr ? reinterpret_cast<const npu_device_fp16_t >(mask_ptr + iq1 mask->get_nb(1) +`
	`109`	`+ (iq3 % mask->get_ne(2)) * mask->get_nb(2)) :`
	`110`	`+ nullptr;`
`109`	`111`
`110`	`112`	`// k indices`
`111`	`113`	`const int ik3 = iq3 / rk3;`