diff --git a/ci/setup_python.env b/ci/setup_python.env index 0305a6329b..d1b4f0336b 100644 --- a/ci/setup_python.env +++ b/ci/setup_python.env @@ -14,7 +14,7 @@ # TVM_FFI_REF=v0.2.0 # Test a specific tag # Uncomment to override TVM-FFI version: -# TVM_FFI_REF= +TVM_FFI_REF=v0.1.10-rc2 # Uncomment to override nvidia-cutlass-dsl version: # CUTLASS_DSL_VERSION= diff --git a/scripts/task_jit_run_tests_part1.sh b/scripts/task_jit_run_tests_part1.sh index 824ed91b96..9e8c22bd8c 100755 --- a/scripts/task_jit_run_tests_part1.sh +++ b/scripts/task_jit_run_tests_part1.sh @@ -13,6 +13,11 @@ if [ "$SKIP_INSTALL" = "0" ]; then pip install -e . -v fi +# Print TVM-FFI version just before running tests +echo "Checking TVM-FFI version before tests..." +python -c "import tvm_ffi; print(f'TVM-FFI version: {tvm_ffi.__version__}')" || true +echo "" + # Run each test file separately to isolate CUDA memory issues pytest -s tests/attention/test_logits_cap.py pytest -s tests/attention/test_sliding_window.py diff --git a/scripts/task_run_unit_tests.sh b/scripts/task_run_unit_tests.sh index e0781d7414..fcf3b44782 100755 --- a/scripts/task_run_unit_tests.sh +++ b/scripts/task_run_unit_tests.sh @@ -11,6 +11,9 @@ SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)" # shellcheck disable=SC1091 # File exists, checked separately source "${SCRIPT_DIR}/test_utils.sh" +# Source test environment setup (handles package overrides like TVM-FFI) +source "${SCRIPT_DIR}/setup_test_env.sh" + # Find and filter test files based on pytest.ini exclusions find_test_files() { echo "Reading pytest.ini for excluded directories..." @@ -84,6 +87,11 @@ main() { # Find test files (unique to unit tests - auto-discovery) find_test_files + # Print TVM-FFI version just before test execution for traceability + echo "Checking TVM-FFI version before tests..." + python -c "import tvm_ffi; print(f'TVM-FFI version: {tvm_ffi.__version__}')" || true + echo "" + # Execute tests or dry run if [ "$DRY_RUN" == "true" ]; then execute_dry_run "$TEST_FILES" diff --git a/scripts/task_test_jit_cache_package_build_import.sh b/scripts/task_test_jit_cache_package_build_import.sh index 1a1dfa56f4..62f83c3ab6 100755 --- a/scripts/task_test_jit_cache_package_build_import.sh +++ b/scripts/task_test_jit_cache_package_build_import.sh @@ -72,6 +72,11 @@ python -c "import torch; print(torch.__version__)" CUDA_VERSION=$(python3 -c 'import torch; print(torch.version.cuda)' | cut -d'.' -f1,2 | tr -d '.') echo "Detected CUDA version: cu${CUDA_VERSION}" +echo "" +echo "Checking TVM-FFI version before verification tests..." +python -c "import tvm_ffi; print(f'TVM-FFI version: {tvm_ffi.__version__}')" || true +echo "" + echo "" echo "========================================" echo "Installing flashinfer package" @@ -108,6 +113,11 @@ echo "✓ Flashinfer-jit-cache wheel installed successfully" cd .. # Verify installation +echo "" +echo "Checking TVM-FFI version before verification tests..." +python -c "import tvm_ffi; print(f'TVM-FFI version: {tvm_ffi.__version__}')" || true +echo "" + echo "" echo "========================================" echo "Running verification tests"