Skip to content

Commit

Permalink
Address reviewer's comments
Browse files Browse the repository at this point in the history
  • Loading branch information
hcho3 committed Oct 8, 2024
1 parent 3efc447 commit 94febb6
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
1 change: 1 addition & 0 deletions qa/L0_e2e/test_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -68,6 +68,7 @@ def valid_shm_modes():


# TODO(hcho3): Remove once we fix the flakiness of CUDA shared mem
# See https://github.com/triton-inference-server/server/issues/7688
def shared_mem_parametrize():
params = [None]
if "cuda" in valid_shm_modes():
Expand Down
3 changes: 1 addition & 2 deletions qa/run_tests.sh
Original file line number Diff line number Diff line change
Expand Up @@ -102,10 +102,9 @@ trap finally EXIT
if [ ! -z $CPU_ONLY ] && [ $CPU_ONLY -eq 1 ]
then
pytest \
-v -rxXs \
--repo "${MODEL_REPO}" \
--hypothesis-profile "$TEST_PROFILE" \
"$QA_DIR"
else
pytest -v -rxXs --repo "${MODEL_REPO}" "$QA_DIR" --hypothesis-profile "$TEST_PROFILE"
pytest --repo "${MODEL_REPO}" "$QA_DIR" --hypothesis-profile "$TEST_PROFILE"
fi

0 comments on commit 94febb6

Please sign in to comment.