ci: Pass FLASHINFER_ENABLE_SM90 to setup.py (#685)

flashinfer-ai · Dec 18, 2024 · 3470329 · 3470329
1 parent 5da7271
commit 3470329
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/run-ci-build-wheel.sh b/scripts/run-ci-build-wheel.sh
@@ -52,7 +52,8 @@ echo "::endgroup::"
 
 echo "::group::Build wheel for FlashInfer"
 cd "$PROJECT_ROOT"
-FLASHINFER_ENABLE_AOT=1 FLASHINFER_LOCAL_VERSION=$FLASHINFER_LOCAL_VERSION python -m build --no-isolation --wheel
+FLASHINFER_ENABLE_AOT=1 FLASHINFER_LOCAL_VERSION=$FLASHINFER_LOCAL_VERSION FLASHINFER_ENABLE_SM90=$FLASHINFER_ENABLE_SM90 \
+    python -m build --no-isolation --wheel
 python -m build --no-isolation --sdist
 ls -la dist/
 echo "::endgroup::"