vllm/vllm/model_executor/parallel_utils at 317b29de0f16428610e2e4d6a6953bee5a2d0ec2 - vllm - Gitea: Git with a cup of tea

biondizzle/vllm

Files

History

Woosuk Kwon a463c333dd Use CuPy for CUDA graphs (#2811 )

2024-02-13 11:32:06 -08:00

..

__init__.py

TP/quantization/weight loading refactor part 1 - Simplify parallel linear logic (#1181 )

2023-10-02 15:36:09 -07:00

communication_op.py

Use CuPy for CUDA graphs (#2811 )

2024-02-13 11:32:06 -08:00

cupy_utils.py

Use CuPy for CUDA graphs (#2811 )

2024-02-13 11:32:06 -08:00

custom_all_reduce.py

[Minor] Fix false warning when TP=1 (#2674 )

2024-01-30 14:39:40 -08:00

parallel_state.py

Use CuPy for CUDA graphs (#2811 )

2024-02-13 11:32:06 -08:00

README.md

Change the name to vLLM (#150 )

2023-06-17 03:07:40 -07:00

utils.py

TP/quantization/weight loading refactor part 2 - Refactor quantized linear logic and extend quantization support to all models (#1622 )

2023-11-15 22:50:41 -08:00

README.md

The files in this folder are ported from Megatron-LM. We only keep the codes that are used in inference.