Skip to content

Commit 0a7f5c7

Browse files
committed
fix to print warning and continue if no jit-cache/cubin available
1 parent 7a27eac commit 0a7f5c7

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

scripts/task_test_blackwell_kernels.sh

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -36,16 +36,14 @@ if [ "$DRY_RUN" != "true" ]; then
3636
echo "Installing flashinfer-cubin from ${DIST_CUBIN_DIR} ..."
3737
pip install -q "${DIST_CUBIN_DIR}"/*.whl
3838
else
39-
echo "ERROR: flashinfer-cubin wheel not found in ${DIST_CUBIN_DIR}. Ensure the CI build stage produced the artifact." >&2
40-
exit 1
39+
echo "WARNING: flashinfer-cubin wheel not found in ${DIST_CUBIN_DIR}. Continuing without precompiled cubins; kernels may JIT-compile during tests." >&2
4140
fi
4241

4342
if [ -d "${DIST_JIT_CACHE_DIR}" ] && ls "${DIST_JIT_CACHE_DIR}"/*.whl >/dev/null 2>&1; then
4443
echo "Installing flashinfer-jit-cache from ${DIST_JIT_CACHE_DIR} ..."
4544
pip install -q "${DIST_JIT_CACHE_DIR}"/*.whl
4645
else
47-
echo "ERROR: flashinfer-jit-cache wheel not found in ${DIST_JIT_CACHE_DIR} for ${CUDA_VERSION}. Ensure the CI build stage produced the artifact." >&2
48-
exit 1
46+
echo "WARNING: flashinfer-jit-cache wheel not found in ${DIST_JIT_CACHE_DIR} for ${CUDA_VERSION}. Continuing without precompiled cache; kernels may JIT-compile during tests." >&2
4947
fi
5048
echo ""
5149

0 commit comments

Comments
 (0)