TensorRT-LLMs/tensorrt_llm/_torch/custom_ops
2025-07-02 04:54:43 -04:00
..
__init__.py
cpp_custom_ops.py [feat] Support torch compile for attention dp (#5086) 2025-07-01 13:48:52 -04:00
flashinfer_custom_ops.py
torch_custom_ops.py [https://nvbugspro.nvidia.com/bug/5329655] [feat] Pytorch path add spec dec param to attention op (#5146) 2025-07-02 04:54:43 -04:00
trtllm_gen_custom_ops.py
userbuffers_custom_ops.py