Fix cutedsl ci error (#20309)
Co-authored-by: Baizhou Zhang <sobereddiezhang@gmail.com>
This commit is contained in:
co-authored by
Baizhou Zhang
parent
456934fed5
commit
680d9d98e4
@@ -38,7 +38,7 @@ dependencies = [
|
|||||||
"msgspec",
|
"msgspec",
|
||||||
"ninja",
|
"ninja",
|
||||||
"numpy",
|
"numpy",
|
||||||
"nvidia-cutlass-dsl>=4.3.4",
|
"nvidia-cutlass-dsl==4.3.5",
|
||||||
"nvidia-ml-py",
|
"nvidia-ml-py",
|
||||||
"openai-harmony==0.0.4",
|
"openai-harmony==0.0.4",
|
||||||
"openai==2.6.1",
|
"openai==2.6.1",
|
||||||
@@ -59,7 +59,7 @@ dependencies = [
|
|||||||
"scipy",
|
"scipy",
|
||||||
"sentencepiece",
|
"sentencepiece",
|
||||||
"setproctitle",
|
"setproctitle",
|
||||||
"sgl-fa4==4.0.3",
|
"sgl-fa4==4.0.4",
|
||||||
"sgl-kernel==0.3.21",
|
"sgl-kernel==0.3.21",
|
||||||
"soundfile==0.13.1",
|
"soundfile==0.13.1",
|
||||||
"tiktoken",
|
"tiktoken",
|
||||||
|
|||||||
@@ -328,6 +328,9 @@ fi
|
|||||||
# Download flashinfer cubins if the local set is incomplete
|
# Download flashinfer cubins if the local set is incomplete
|
||||||
bash "${SCRIPT_DIR}/ci_download_flashinfer_cubin.sh"
|
bash "${SCRIPT_DIR}/ci_download_flashinfer_cubin.sh"
|
||||||
|
|
||||||
|
# Clean nvidia-cutlass-dsl-libs-base for cutedsl lower than 0.4.4
|
||||||
|
$PIP_UNINSTALL_CMD nvidia-cutlass-dsl-libs-base $PIP_UNINSTALL_SUFFIX || true
|
||||||
|
|
||||||
# Show current packages
|
# Show current packages
|
||||||
$PIP_CMD list
|
$PIP_CMD list
|
||||||
python3 -c "import torch; print(torch.version.cuda)"
|
python3 -c "import torch; print(torch.version.cuda)"
|
||||||
|
|||||||
Reference in New Issue
Block a user