From 9ea2c9ec55233eedc94ce25df1edff0d23e693be Mon Sep 17 00:00:00 2001
From: root <cuikuilong.ckl@alibaba-inc.com>
Date: Tue, 5 Nov 2024 08:32:13 +0000
Subject: [PATCH] fix

---
 Makefile                          |  8 -----
 tests/unit_test/queue/test_zmq.py |  2 +-
 tools/cupy_cuda_install.sh        | 52 -------------------------------
 3 files changed, 1 insertion(+), 61 deletions(-)
 delete mode 100755 tools/cupy_cuda_install.sh

diff --git a/Makefile b/Makefile
index 6bc87a9b..8c2a11fc 100644
--- 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
index d4303d37..052dad52 100644
--- 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
deleted file mode 100755
index bfd09cf8..00000000
--- a/tools/cupy_cuda_install.sh
+++ /dev/null
@@ -1,52 +0,0 @@
-# Copyright (c) 2024, Alibaba Group;
-# Licensed under the Apache License, Version 2.0 (the "License");
-# you may not use this file except in compliance with the License.
-# You may obtain a copy of the License at
-
-# http://www.apache.org/licenses/LICENSE-2.0
-
-# Unless required by applicable law or agreed to in writing, software
-# distributed under the License is distributed on an "AS IS" BASIS,
-# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-# See the License for the specific language governing permissions and
-# limitations under the License.
-
-#!/bin/bash
-
-if ! command -v nvcc &> /dev/null
-then
-    echo "CUDA is not installed."
-    exit 1
-fi
-
-CUDA_VERSION=$(nvcc --version | grep -oP 'V\d+\.\d+' | head -1 | tr -d 'V')
-
-case $CUDA_VERSION in
-    10.0)
-        pip install cupy-cuda100
-        ;;
-    10.1)
-        pip install cupy-cuda101
-        ;;
-    10.2)
-        pip install cupy-cuda102
-        ;;
-    11.0)
-        pip install cupy-cuda110
-        ;;
-    11.1)
-        pip install cupy-cuda111
-        ;;
-    11.2 | 11.3 | 11.4)
-        pip install cupy-cuda11x
-        ;;
-    12.*)
-        pip install cupy-cuda12x
-        ;;
-    *)
-        echo "Detected CUDA version: $CUDA_VERSION, please refer to https://pypi.org/search/?o=&q=cupy-cuda&page=1 and the CuPy documentation for supported versions."
-        exit 1
-        ;;
-esac
-
-echo "CuPy for CUDA $CUDA_VERSION installed successfully."