TensorRT-LLMs/cpp/tests/unit_tests/kernels
Cheng Hang 64db7d27f6
[feat] Optimizations on weight-only batched gemv kernel (#5420)
Signed-off-by: Cheng Hang <chang@nvidia.com>
2025-06-30 10:20:16 +08:00
..
allReduce
cudaCoreGemm
fused_gated_gemm
routing
sampling
smoothQuant
weightOnly [feat] Optimizations on weight-only batched gemv kernel (#5420) 2025-06-30 10:20:16 +08:00
banRepeatNGramsKernelsTest.cpp
CMakeLists.txt
decodingKernelTest.cpp
logitsBitmaskTest.cpp
mixtureOfExpertsTest.cu
mlaChunkedPrefillTest.cu
mlaPreprocessTest.cu
ropeTest.cu
shiftKCacheKernelTest.cu
stopCriteriaKernelsTest.cpp