diff --git a/.ci/tritonbench/test-gpu.sh b/.ci/tritonbench/test-gpu.sh index 91085022..e8d22859 100644 --- a/.ci/tritonbench/test-gpu.sh +++ b/.ci/tritonbench/test-gpu.sh @@ -11,4 +11,7 @@ fi # install deps pip install psutil tabulate +# FIXME: patch hstu +python install.py --hstu + python -m unittest test.test_gpu.main -v