Dockerfile.rocm · 0258b7a94b08321ca01cf170f867b67c1920af87 · OpenDAS / vllm_cscc

[ROCm][Hardware][AMD] Use Triton Kernel for default FA on ROCm (#3643) · 6c0b0451

Juan Villamizar authored Apr 09, 2024


Co-authored-by: jpvillam <jpvillam@amd.com>
Co-authored-by: Gregory Shtrasberg <Gregory.Shtrasberg@amd.com>
Co-authored-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>

6c0b0451

Dockerfile.rocm 3.42 KB

Replace Dockerfile.rocm