Skip to content

Commit bc3b93c

Browse files
limin2021claude
andcommitted
fix: CUDA 12 needs no extra, only CUDA 13 needs [cu13] for cutlass-dsl
- setup_test_env.sh: only add [cu13] extra for CUDA 13, plain install for CUDA 12 - Update skill docs to match NVIDIA's official installation guide Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent ae215a6 commit bc3b93c

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

scripts/setup_test_env.sh

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -26,19 +26,19 @@ fi
2626

2727
# Override nvidia-cutlass-dsl if specified
2828
if [ -n "${CUTLASS_DSL_VERSION:-}" ]; then
29-
# Detect CUDA major version to select the correct extra (cu12 or cu13)
29+
# Detect CUDA major version: only CUDA 13+ needs [cu13] extra
3030
CUDA_MAJOR=$(python -c "import torch; print(torch.version.cuda.split('.')[0])" 2>/dev/null || echo "12")
3131
if [ "$CUDA_MAJOR" = "13" ]; then
32-
CUTLASS_DSL_EXTRA="cu13"
32+
CUTLASS_DSL_PKG="nvidia-cutlass-dsl[cu13]==${CUTLASS_DSL_VERSION}"
3333
else
34-
CUTLASS_DSL_EXTRA="cu12"
34+
CUTLASS_DSL_PKG="nvidia-cutlass-dsl==${CUTLASS_DSL_VERSION}"
3535
fi
3636
echo "========================================"
37-
echo "Overriding nvidia-cutlass-dsl with version: ${CUTLASS_DSL_VERSION} [${CUTLASS_DSL_EXTRA}]"
37+
echo "Overriding nvidia-cutlass-dsl with: ${CUTLASS_DSL_PKG}"
3838
echo "========================================"
3939
# Clean uninstall old packages first (recommended by NVIDIA docs)
4040
pip uninstall nvidia-cutlass-dsl nvidia-cutlass-dsl-libs-base nvidia-cutlass-dsl-libs-cu12 nvidia-cutlass-dsl-libs-cu13 -y 2>/dev/null || true
41-
pip install "nvidia-cutlass-dsl[${CUTLASS_DSL_EXTRA}]==${CUTLASS_DSL_VERSION}"
41+
pip install "${CUTLASS_DSL_PKG}"
4242
echo "nvidia-cutlass-dsl override complete."
4343
echo ""
4444
fi

0 commit comments

Comments
 (0)