TensorRT-LLMs/tensorrt_llm/layers
yunruis 30c5b4183a
refactoring: port customized kernels with public cutlass version (#5027)
Signed-off-by: yunruis 

Merge this to unblock others since the full CI has been run through
2025-06-13 16:19:31 +08:00
..
__init__.py
activation.py
attention.py refactor: use x is None instead of x == None. (#4244) 2025-05-15 20:00:04 +08:00
cast.py
conv.py
embedding.py
language_adapter.py
linear.py
lora.py
mlp.py
moe.py refactoring: port customized kernels with public cutlass version (#5027) 2025-06-13 16:19:31 +08:00
normalization.py
pooling.py
recurrent.py
ssm.py