..
allgatherOp.cpp
Update TensorRT-LLM ( #2792 )
2025-02-18 21:27:39 +08:00
allreduceOp.cpp
feat: Introduce UB allocator for pytorch flow ( #3257 )
2025-04-08 18:39:49 +08:00
attentionOp.cpp
feat: Add support for FP8 MLA on Hopper and Blackwell. ( #3190 )
2025-04-07 15:14:13 +08:00
CMakeLists.txt
feat: trtllm-gen fp4 GEMM for pytorch workflow ( #3423 )
2025-04-11 02:28:07 +08:00
convertSpecDecodingMaskToPackedMaskOp.cpp
Update TensorRT-LLM ( #2873 )
2025-03-11 21:13:42 +08:00
cublasScaledMM.cpp
feat: Introduce UB allocator for pytorch flow ( #3257 )
2025-04-08 18:39:49 +08:00
cutlassScaledMM.cpp
Update TensorRT-LLM ( #2582 )
2024-12-16 21:50:47 -08:00
deepseekAllreduceFusionOp.cpp
feat: fix and improve allreduce and fusion kernels ( #3064 )
2025-04-08 19:33:52 +08:00
dynamicDecodeOp.cpp
Feat: Variable-Beam-Width-Search (VBWS) part3 ( #3338 )
2025-04-08 23:51:27 +08:00
dynamicDecodeOp.h
Update TensorRT-LLM ( #2783 )
2025-02-13 18:40:22 +08:00
fmhaPackMaskOp.cpp
Update TensorRT-LLM ( #2873 )
2025-03-11 21:13:42 +08:00
fp4BatchedQuantize.cpp
update FP4 quantize layout ( #3045 )
2025-04-03 13:13:54 -04:00
fp4Gemm.cpp
feat: Apply the new torch-flow compatible AutoTuner to both Fused MoE and NVFP4 Linear operators. ( #3151 )
2025-04-08 14:28:36 +08:00
fp4GemmTrtllmGen.cpp
feat: trtllm-gen fp4 GEMM for pytorch workflow ( #3423 )
2025-04-11 02:28:07 +08:00
fp4Op.cpp
update FP4 quantize layout ( #3045 )
2025-04-03 13:13:54 -04:00
fp4Quantize.cpp
update FP4 quantize layout ( #3045 )
2025-04-03 13:13:54 -04:00
fp8BlockScaleMoe.cpp
Update TensorRT-LLM ( #2873 )
2025-03-11 21:13:42 +08:00
fp8BlockScalingGemm.cpp
[feat] open source fp8_blockscale_gemm ( #3071 )
2025-04-02 12:12:52 +08:00
fp8Op.cpp
Update TensorRT-LLM ( #2755 )
2025-02-11 03:01:00 +00:00
fp8Quantize.cpp
[feat] open source fp8_blockscale_gemm ( #3071 )
2025-04-02 12:12:52 +08:00
fusedTopkSoftmax.cpp
Add Llama 4 ( #3302 )
2025-04-09 03:35:21 +08:00
gatherTreeOp.cpp
Update TensorRT-LLM ( #2873 )
2025-03-11 21:13:42 +08:00
logitsBitmaskOp.cpp
Update ( #2978 )
2025-03-23 16:39:35 +08:00
mambaConv1dOp.cpp
Update TensorRT-LLM ( #2873 )
2025-03-11 21:13:42 +08:00
moeOp.cpp
feat: Apply the new torch-flow compatible AutoTuner to both Fused MoE and NVFP4 Linear operators. ( #3151 )
2025-04-08 14:28:36 +08:00
mtpOp.cpp
Update TensorRT-LLM ( #2849 )
2025-03-04 18:44:00 +08:00
ncclCommunicatorOp.cpp
Update TensorRT-LLM ( #941 )
2024-01-23 23:22:35 +08:00
ncclCommunicatorOp.h
Update TensorRT-LLM ( #787 )
2024-01-02 17:54:32 +08:00
noAuxTcOp.cpp
Update TensorRT-LLM ( #2873 )
2025-03-11 21:13:42 +08:00
parallelDecodeKVCacheUpdateOp.cpp
Update TensorRT-LLM ( #2582 )
2024-12-16 21:50:47 -08:00
redrafterCurandOp.cpp
Update TensorRT-LLM ( #2873 )
2025-03-11 21:13:42 +08:00
reducescatterOp.cpp
Update TensorRT-LLM ( #2792 )
2025-02-18 21:27:39 +08:00
relativeAttentionBiasOp.cpp
Update TensorRT-LLM ( #2873 )
2025-03-11 21:13:42 +08:00
selectiveScanOp.cpp
Update TensorRT-LLM ( #2873 )
2025-03-11 21:13:42 +08:00
thUtils.cpp
Update TensorRT-LLM ( #2849 )
2025-03-04 18:44:00 +08:00
thUtils.h
Update TensorRT-LLM ( #2849 )
2025-03-04 18:44:00 +08:00
userbuffersFinalizeOp.cpp
feat: Introduce UB allocator for pytorch flow ( #3257 )
2025-04-08 18:39:49 +08:00
userbuffersTensor.cpp
feat: Introduce UB allocator for pytorch flow ( #3257 )
2025-04-08 18:39:49 +08:00
userbuffersTensor.h
feat: Introduce UB allocator for pytorch flow ( #3257 )
2025-04-08 18:39:49 +08:00
weightOnlyQuantOp.cpp
chore: remove usernames from comments ( #3291 )
2025-04-05 13:44:28 +08:00
a8310b01dc