diff --git a/kernels/load_and_reshape_flash.cpp b/kernels/load_and_reshape_flash.cpp
index 8117feeb67392a93734c9ffb6cb78ba54e9eb4c2..4700f4a16e9317cd4434504f211d6a57ba61be1d 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,