From d580cc68f0b57446e2aff1e7caf14c63161c6302 Mon Sep 17 00:00:00 2001
From: G F M Yeung <gingfung.matthew.yeung@huawei.com>
Date: Fri, 15 Aug 2025 06:36:03 +0000
Subject: [PATCH] update kernels/load_and_reshape_flash.cpp. - removed unused
 problematic macros

Signed-off-by: G F M Yeung <gingfung.matthew.yeung@huawei.com>
---
 kernels/load_and_reshape_flash.cpp | 2 --
 1 file changed, 2 deletions(-)

diff --git a/kernels/load_and_reshape_flash.cpp b/kernels/load_and_reshape_flash.cpp
index 8117fee..4700f4a 100644
--- a/kernels/load_and_reshape_flash.cpp
+++ b/kernels/load_and_reshape_flash.cpp
@@ -239,8 +239,6 @@ extern void load_and_reshape_flash_kernel(kvcache_ops::AscendType type, kvcache_
                             const int32_t pagedSize, const int32_t numTokens, const int32_t numLayers,
                             const int32_t layerIdx, bool page2L)
 {
-    KERNEL_TASK_TYPE_DEFAULT(KERNEL_TYPE_AIV_ONLY);
-
     switch(type) {
         case kvcache_ops::AscendType::FP16:
             dispatch_on_slot_type<half>(slotType, blockDim, stream, dstCacheTensor, keyCachePtr, valueCachePtr, 
-- 
Gitee