From f40a0969ce6d8d7f31135168ad5145aa1edec20d Mon Sep 17 00:00:00 2001
From: Xu Zhao <xzhao9@meta.com>
Date: Fri, 8 Nov 2024 20:02:20 -0500
Subject: [PATCH] Add pr ci

---
 .ci/tritonbench/test-gpu.sh | 3 +++
 requirements.txt            | 2 ++
 2 files changed, 5 insertions(+)

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