Skip to content

Conversation

@maleksan85
Copy link

Wrapping kernels that has incompatibility with Navi arch with conditional compilation to unblock Navi execution.

Example command to run on Navi is: VLLM_USE_ROCM_CUSTOM_PAGED_ATTN=0 VLLM_USE_ROCM_SKINNY_GEMM=0 python benchmarks/benchmark_latency.py --model /data/models/llama-2-7b-chat-hf --input-len 2048 --output-len 128 --batch-size 8

TODO port kernels to Navi.

gshtras
gshtras previously approved these changes Aug 5, 2024
Aleksandr Malyshev added 2 commits August 5, 2024 22:09
@maleksan85 maleksan85 merged commit c034d5d into main Aug 6, 2024
@gshtras gshtras deleted the cl/fix-navi-build branch August 8, 2024 17:34
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants