We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c9484db commit ea2f9e3Copy full SHA for ea2f9e3
.github/workflows/build-vllm.yaml
@@ -51,6 +51,8 @@ jobs:
51
build-args: |
52
CUDA_VERSION=${{ matrix.cuda_version }}
53
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 }}
56
TRITON_REF=${{ env.TRITON_REF }}
57
TRITON_BUILD_VERSION=${{ env.TRITON_BUILD_VERSION }}
58
XFORMERS_REF=${{ env.XFORMERS_REF }}
0 commit comments