Skip to content

Commit ea2f9e3

Browse files
committed
Actually thread archs into build args
1 parent c9484db commit ea2f9e3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

.github/workflows/build-vllm.yaml

+2
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,8 @@ jobs:
5151
build-args: |
5252
CUDA_VERSION=${{ matrix.cuda_version }}
5353
IMAGE_DISTRO=${{ matrix.image_distro }}
54+
TORCH_CUDA_ARCH_LIST=${{ env.TORCH_CUDA_ARCH_LIST }}
55+
VLLM_FA_CMAKE_GPU_ARCHES=${{ env.VLLM_FA_CMAKE_GPU_ARCHES }}
5456
TRITON_REF=${{ env.TRITON_REF }}
5557
TRITON_BUILD_VERSION=${{ env.TRITON_BUILD_VERSION }}
5658
XFORMERS_REF=${{ env.XFORMERS_REF }}

0 commit comments

Comments
 (0)