Fix missing packet parameter for executor (#385)

microsoft · Nov 19, 2024 · 3e51e9b · 3e51e9b
1 parent b3dc74c
commit 3e51e9b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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)
           ,