vllm/csrc/punica
Joe Runde ba4994443a
[Kernel] Add punica dimensions for Granite 3b and 8b (#5930)
Signed-off-by: Joe Runde <joe@joerun.de>
2024-06-29 10:48:25 +08:00
..
bgmv [Kernel] Add punica dimensions for Granite 3b and 8b (#5930) 2024-06-29 10:48:25 +08:00
LICENSE [Experimental] Add multi-LoRA support (#1804) 2024-01-23 15:26:37 -08:00
punica_ops.cu [Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047) 2024-06-09 16:23:30 -04:00
punica_ops.h [Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047) 2024-06-09 16:23:30 -04:00
torch_bindings.cpp [Kernel][Misc] Use TORCH_LIBRARY instead of PYBIND11_MODULE for custom ops (#5047) 2024-06-09 16:23:30 -04:00
type_convert.h [ROCm] Add support for Punica kernels on AMD GPUs (#3140) 2024-05-09 09:19:50 -07:00