|
|
|
|
|
|
|
#include "kernel.h" |
|
#include "marlin_template.h" |
|
|
|
namespace MARLIN_NAMESPACE_NAME { |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 256, 1, 8, 8, true, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 1, 8, 4, true, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 1, 4, 8, true, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 256, 1, 8, 8, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 1, 8, 4, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 1, 4, 8, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 256, 2, 16, 4, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 2, 8, 4, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 2, 4, 8, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 256, 3, 16, 4, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 3, 8, 4, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 3, 4, 8, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 256, 4, 16, 4, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 4, 8, 4, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
template __global__ void Marlin<half, vllm::kFE2M1f.id(), 128, 4, 4, 8, false, pipe_stages, 1, false>( MARLIN_KERNEL_PARAMS ); |
|
|
|
} |
|
|