Woosuk Kwon 8267b06c30 [Kernel] Add Triton MoE kernel configs for DBRX on A100 (#3679)

2024-03-27 22:22:25 -07:00

..

fused_moe

[Kernel] Add Triton MoE kernel configs for DBRX on A100 (#3679)

2024-03-27 22:22:25 -07:00

ops

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

quantization

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

__init__.py

Change the name to vLLM (#150)

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

activation.py

[1/n][Chunked Prefill] Refactor input query shapes (#3236)

2024-03-20 14:46:05 -07:00

layernorm.py

Revert "Refactor llama family models (#2637)" (#2851)

2024-02-13 09:24:59 -08:00

linear.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

logits_processor.py

[Hardware][Neuron] Refactor neuron support (#3471)

2024-03-22 01:22:17 +00:00

rejection_sampler.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00

rotary_embedding.py

[BugFix] tensor.get_device() -> tensor.device (#3604)

2024-03-24 19:01:13 -07:00

sampler.py

Optimize _get_ranks in Sampler (#3623)

2024-03-25 16:03:02 -07:00

vocab_parallel_embedding.py

[CI] Try introducing isort. (#3495)

2024-03-25 07:59:47 -07:00