Skip to content

Commit f921ead

Browse files
mc-nvyinggeh
andauthored
TPRD-1710: Update default branches post-25.09 (#99)
Co-authored-by: Yingge He <[email protected]>
1 parent 2d0f250 commit f921ead

File tree

6 files changed

+6
-4
lines changed

6 files changed

+6
-4
lines changed

ci/L0_additional_outputs_vllm/test.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,6 @@ source ../common/util.sh
3030

3131
pip3 install pytest==8.1.1
3232
pip3 install tritonclient[grpc]
33-
pip3 install "transformers<=4.53.3" # TODO:DLIS-8441 remove this dependency
3433

3534
# Prepare Model
3635
rm -rf models vllm_baseline_output.pkl && mkdir -p models

ci/L0_backend_vllm/test.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ RET=0
2929
SUBTESTS="accuracy_test request_cancellation enabled_stream vllm_backend metrics_test"
3030

3131
python3 -m pip install tritonclient[grpc]
32-
python3 -m pip install "transformers<=4.53.3" # TODO:DLIS-8441 remove this dependency
3332

3433
for TEST in ${SUBTESTS}; do
3534
(cd ${TEST} && bash -ex test.sh && cd ..)

ci/L0_check_health_vllm/test.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,6 @@ source ../common/util.sh
3030

3131
pip3 install pytest==8.1.1
3232
pip3 install tritonclient[grpc]
33-
pip3 install "transformers<=4.53.3" # TODO:DLIS-8441 remove this dependency
3433

3534
RET=0
3635

ci/L0_multi_gpu_vllm/multi_lora/test.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -41,6 +41,9 @@ EXPECTED_NUM_TESTS=2
4141
GENERATE_ENDPOINT="localhost:8000/v2/models/vllm_llama_multi_lora/generate"
4242
CHECK_FOR_ERROR=true
4343

44+
export C_INCLUDE_PATH=/usr/local/cuda/include:$C_INCLUDE_PATH
45+
export TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas
46+
4447
make_api_call() {
4548
local endpoint="$1"
4649
local data="$2"

ci/L0_multi_gpu_vllm/test.sh

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ RET=0
2929
SUBTESTS="vllm_backend multi_lora"
3030

3131
python3 -m pip install tritonclient[grpc]
32-
python3 -m pip install "transformers<=4.53.3" # TODO:DLIS-8441 remove this dependency
3332

3433
for TEST in ${SUBTESTS}; do
3534
(cd ${TEST} && bash -ex test.sh && cd ..)

ci/L0_multi_gpu_vllm/vllm_backend/test.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,6 +36,9 @@ CLIENT_PY="./vllm_multi_gpu_test.py"
3636
SAMPLE_MODELS_REPO="../../../samples/model_repository"
3737
EXPECTED_NUM_TESTS=1
3838

39+
export C_INCLUDE_PATH=/usr/local/cuda/include:$C_INCLUDE_PATH
40+
export TRITON_PTXAS_PATH=/usr/local/cuda/bin/ptxas
41+
3942
### Helpers
4043
function validate_file_contains() {
4144
local KEY="${1}"

0 commit comments

Comments
 (0)