TensorRT-LLMs/cpp/tensorrt_llm/kernels
2023-10-18 22:18:19 +08:00
..
contextFusedMultiHeadAttention refresh 0.5.0 release branch with the latest revision 2023-10-18 22:18:19 +08:00
cutlass_kernels refresh 0.5.0 release branch with the latest revision 2023-10-18 22:18:19 +08:00
decoderMaskedMultiheadAttention Updates for release/0.5.0 2023-10-15 21:26:20 +08:00
onlineSoftmaxBeamsearchKernels Updates for release/0.5.0 2023-10-15 21:26:20 +08:00
weightOnlyBatchedGemv Updates for release/0.5.0 2023-10-15 21:26:20 +08:00
banBadWords.cu Update code 2023-09-28 09:00:05 -07:00
banBadWords.h Initial commit 2023-09-20 00:29:41 -07:00
banRepeatNgram.cu Update code 2023-09-28 09:00:05 -07:00
banRepeatNgram.h Initial commit 2023-09-20 00:29:41 -07:00
beamSearchPenaltyKernels.cu Update code 2023-09-28 09:00:05 -07:00
beamSearchPenaltyKernels.h Update code 2023-09-28 09:00:05 -07:00
beamSearchTopkKernels.cu Update code 2023-09-28 09:00:05 -07:00
beamSearchTopkKernels.h Initial commit 2023-09-20 00:29:41 -07:00
CMakeLists.txt Initial commit 2023-09-20 00:29:41 -07:00
customAllReduceKernels.cu Updates for release/0.5.0 2023-10-15 21:26:20 +08:00
customAllReduceKernels.h Updates for release/0.5.0 2023-10-15 21:26:20 +08:00
decoderMaskedMultiheadAttention.cu refresh 0.5.0 release branch with the latest revision 2023-10-18 22:18:19 +08:00
decoderMaskedMultiheadAttention.h refresh 0.5.0 release branch with the latest revision 2023-10-18 22:18:19 +08:00
decoderMaskedMultiheadAttentionUtils.h Update 2023-10-10 23:22:17 -07:00
decodingKernels.cu refresh 0.5.0 release branch with the latest revision 2023-10-18 22:18:19 +08:00
decodingKernels.h refresh 0.5.0 release branch with the latest revision 2023-10-18 22:18:19 +08:00
gptKernels.cu Initial commit 2023-09-20 00:29:41 -07:00
gptKernels.h Updates for release/0.5.0 2023-10-15 21:26:20 +08:00
kvCacheUtils.h Initial commit 2023-09-20 00:29:41 -07:00
layernormKernels.cu Initial commit 2023-09-20 00:29:41 -07:00
layernormKernels.h Initial commit 2023-09-20 00:29:41 -07:00
lookupKernels.cu Initial commit 2023-09-20 00:29:41 -07:00
lookupKernels.h Initial commit 2023-09-20 00:29:41 -07:00
onlineSoftmaxBeamsearchKernels.cu Update code 2023-09-28 09:00:05 -07:00
onlineSoftmaxBeamsearchKernels.h Initial commit 2023-09-20 00:29:41 -07:00
penaltyTypes.h Initial commit 2023-09-20 00:29:41 -07:00
preQuantScaleKernel.cu Updates for release/0.5.0 2023-10-15 21:26:20 +08:00
preQuantScaleKernel.h Initial commit 2023-09-20 00:29:41 -07:00
quantization.cu Update code 2023-09-28 09:00:05 -07:00
quantization.h Initial commit 2023-09-20 00:29:41 -07:00
rmsnormKernels.cu Initial commit 2023-09-20 00:29:41 -07:00
rmsnormKernels.h Initial commit 2023-09-20 00:29:41 -07:00
samplingPenaltyKernels.cu Update code 2023-09-28 09:00:05 -07:00
samplingPenaltyKernels.h Update code 2023-09-28 09:00:05 -07:00
samplingTopKKernels.cu Update code 2023-09-28 09:00:05 -07:00
samplingTopKKernels.h Initial commit 2023-09-20 00:29:41 -07:00
samplingTopPKernels.cu Initial commit 2023-09-20 00:29:41 -07:00
samplingTopPKernels.h Initial commit 2023-09-20 00:29:41 -07:00
stopCriteriaKernels.cu Update 2023-10-10 23:22:17 -07:00
stopCriteriaKernels.h Initial commit 2023-09-20 00:29:41 -07:00
unfusedAttentionKernels_2.cu refresh 0.5.0 release branch with the latest revision 2023-10-18 22:18:19 +08:00
unfusedAttentionKernels.cu refresh 0.5.0 release branch with the latest revision 2023-10-18 22:18:19 +08:00
unfusedAttentionKernels.h Updates for release/0.5.0 2023-10-15 21:26:20 +08:00