diff --git a/.ci/tritonbench/test-gpu.sh b/.ci/tritonbench/test-gpu.sh index 0f786130..bf762c37 100644 --- a/.ci/tritonbench/test-gpu.sh +++ b/.ci/tritonbench/test-gpu.sh @@ -8,4 +8,7 @@ fi . "${SETUP_SCRIPT}" +# install deps +pip install psutil tabulate + python -m unittest test.test_gpu.main diff --git a/requirements.txt b/requirements.txt index ae8d8208..1cc03e35 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,3 +1,5 @@ packaging pynvml +psutil +tabulate transformers==4.46.1