fix

AlibabaPAI · Nov 5, 2024 · 9ea2c9e · 9ea2c9e
1 parent 7b13817
commit 9ea2c9e
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 61 deletions.
diff --git a/Makefile b/Makefile
@@ -65,14 +65,6 @@ pygloo: init
 
 ##################### pygloo install for gloo migration backend end #####################
 
-###################################### cupy begin #######################################
-
-.PHONY: cupy-cuda
-cupy-cuda:
-	@./tools/cupy_cuda_install.sh
-
-####################################### cupy end ########################################
-
 ##################################### pylint begin ######################################
 
 PYLINT_VERSION = 2.12.2

diff --git a/tests/unit_test/queue/test_zmq.py b/tests/unit_test/queue/test_zmq.py
@@ -106,8 +106,8 @@ async def benchmark_queue(qps, ip=None, port=None):
         signal.alarm(0)
 
 @pytest.mark.asyncio
-@pytest.mark.parametrize("qps", [128.0, 256.0, 512.0, 1024.0])
 async def test_queue_zmq(setup_ray_env, qps):
     ip = '127.0.0.1'
     port = 1234
+    qps = 1024.0
     await benchmark_queue(qps, ip, port)
diff --git a/tools/cupy_cuda_install.sh b/tools/cupy_cuda_install.sh