Skip to content

Commit

Permalink
xformat
Browse files Browse the repository at this point in the history
  • Loading branch information
LucasWilkinson committed Sep 26, 2024
1 parent 94613cd commit 8d4597f
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -238,7 +238,7 @@ if(VLLM_GPU_LANG STREQUAL "CUDA")
CUDA_ARCHS "${CUDA_ARCHS}"
BUILD_PTX_FOR_HIGHEST_ARCH)

# Only build marlin if we are building for atleast some compatible archs
# Only build marlin if we are building for at least some compatible archs
get_minimal_compatible_cuda_archs("8.0;8.6;8.9")
if (COMPATIBLE_ARCHS)
set(MARLIN_SRCS
Expand Down
2 changes: 1 addition & 1 deletion tests/kernels/test_marlin_kernel.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import vllm._custom_ops
import vllm._custom_ops # noqa: F401
import torch
import pytest

Expand Down

0 comments on commit 8d4597f

Please sign in to comment.