From b9ec407610c26b75d7c5d75700ead9fc0afce1c4 Mon Sep 17 00:00:00 2001
From: Ziyue Yang <ziyyang@microsoft.com>
Date: Sun, 17 Nov 2024 03:58:39 +0000
Subject: [PATCH] Fix missing packet parameter for executor

---
 src/executor/execution_kernel.cu | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/executor/execution_kernel.cu b/src/executor/execution_kernel.cu
index ed0fdc505..a60317c51 100644
--- a/src/executor/execution_kernel.cu
+++ b/src/executor/execution_kernel.cu
@@ -22,7 +22,7 @@ void ExecutionKernel::launchKernel(int rank, int nthreadblocks, int nthreads, vo
 #endif
       break;
     case DataType::UINT32:
-      executionKernel<uint32_t><<<nthreadblocks, nthreads, sharedMemSize, stream>>>(
+      executionKernel<uint32_t, PacketType><<<nthreadblocks, nthreads, sharedMemSize, stream>>>(
           rank, (uint32_t*)src, (uint32_t*)dst, (uint32_t*)scratch, scratchSize, plan, flag
 #if defined(ENABLE_NPKIT)
           ,
@@ -32,7 +32,7 @@ void ExecutionKernel::launchKernel(int rank, int nthreadblocks, int nthreads, vo
 #endif
       break;
     case DataType::FLOAT16:
-      executionKernel<half><<<nthreadblocks, nthreads, sharedMemSize, stream>>>(
+      executionKernel<half, PacketType><<<nthreadblocks, nthreads, sharedMemSize, stream>>>(
           rank, (half*)src, (half*)dst, (half*)scratch, scratchSize, plan, flag
 #if defined(ENABLE_NPKIT)
           ,
@@ -42,7 +42,7 @@ void ExecutionKernel::launchKernel(int rank, int nthreadblocks, int nthreads, vo
 #endif
       break;
     case DataType::FLOAT32:
-      executionKernel<float><<<nthreadblocks, nthreads, sharedMemSize, stream>>>(
+      executionKernel<float, PacketType><<<nthreadblocks, nthreads, sharedMemSize, stream>>>(
           rank, (float*)src, (float*)dst, (float*)scratch, scratchSize, plan, flag
 #if defined(ENABLE_NPKIT)
           ,
@@ -52,7 +52,7 @@ void ExecutionKernel::launchKernel(int rank, int nthreadblocks, int nthreads, vo
 #endif
       break;
     case DataType::BFLOAT16:
-      executionKernel<__bfloat16><<<nthreadblocks, nthreads, sharedMemSize, stream>>>(
+      executionKernel<__bfloat16, PacketType><<<nthreadblocks, nthreads, sharedMemSize, stream>>>(
           rank, (__bfloat16*)src, (__bfloat16*)dst, (__bfloat16*)scratch, scratchSize, plan, flag
 #if defined(ENABLE_NPKIT)
           ,