TensorRT-LLMs/cpp/tensorrt_llm/kernels/trtllmGenKernels/blockScaleMoe
ChristinaZ 12d8c7d129
Refactor the topk parallelization part for the routing kernels (#5567)
Signed-off-by: Christina Zhang <83400082+ChristinaZ@users.noreply.github.com>
2025-07-07 15:53:25 +08:00
..
CMakeLists.txt feat: [Deepseek] Add trtllm-gen MOE FP4 MOE backend (#3387) 2025-04-21 10:01:33 +08:00
DevKernel.cu feat: update DeepSeek FP8 TRT-LLM Gen cubins (#4643) 2025-06-03 14:07:54 -07:00
DevKernel.h Refactor the topk parallelization part for the routing kernels (#5567) 2025-07-07 15:53:25 +08:00
IntFastDiv.h [fix] Fix comment to pass guardwords check (#5191) 2025-06-13 15:49:59 +08:00
RoutingKernel.cu Refactor the topk parallelization part for the routing kernels (#5567) 2025-07-07 15:53:25 +08:00
RoutingKernel.h Refactor the topk parallelization part for the routing kernels (#5567) 2025-07-07 15:53:25 +08:00
RoutingKernelTopK.cuh Refactor the topk parallelization part for the routing kernels (#5567) 2025-07-07 15:53:25 +08:00
runner.cu Refactor the topk parallelization part for the routing kernels (#5567) 2025-07-07 15:53:25 +08:00
runner.h [TRTLLM-5770] feat: Integrate TRT-LLM Gen FP8 block scale MoE with Pytorch workflow kernel autotuner (#5207) 2025-06-17 21:01:56 +08:00