From 8f638e609ccf31368c9c0b6d38b6d2f297b32894 Mon Sep 17 00:00:00 2001
From: YanxuanLiu <yanxuanl@nvidia.com>
Date: Fri, 8 Dec 2023 13:28:46 +0800
Subject: [PATCH] remove cuda version

Signed-off-by: YanxuanLiu <yanxuanl@nvidia.com>
---
 .github/workflows/gcs-benchmark.yml        | 1 -
 python/benchmark/dataproc/start_cluster.sh | 2 --
 2 files changed, 3 deletions(-)

diff --git a/.github/workflows/gcs-benchmark.yml b/.github/workflows/gcs-benchmark.yml
index 9ef57ac6..8802e387 100644
--- a/.github/workflows/gcs-benchmark.yml
+++ b/.github/workflows/gcs-benchmark.yml
@@ -27,7 +27,6 @@ on:
 
 jobs:
   Benchmark:
-    if: github.repository == 'NVIDIA/spark-rapids-ml'
     runs-on: ubuntu-latest
     env:
       PROJECT: rapids-spark
diff --git a/python/benchmark/dataproc/start_cluster.sh b/python/benchmark/dataproc/start_cluster.sh
index 49ae1a4b..76fa11a1 100755
--- a/python/benchmark/dataproc/start_cluster.sh
+++ b/python/benchmark/dataproc/start_cluster.sh
@@ -13,7 +13,6 @@ if [[ -z ${GCS_BUCKET} ]]; then
 fi
 
 BENCHMARK_HOME=${BENCHMARK_HOME:-${GCS_BUCKET}/benchmark}
-CUDA_VERSION=${CUDA_VERSION:-12.2.2}
 
 gpu_args=$(cat <<EOF
 --master-accelerator type=nvidia-tesla-t4,count=1
@@ -21,7 +20,6 @@ gpu_args=$(cat <<EOF
 --initialization-actions gs://${BENCHMARK_HOME}/spark-rapids.sh,gs://${BENCHMARK_HOME}/init_benchmark.sh
 --metadata gpu-driver-provider="NVIDIA"
 --metadata rapids-runtime=SPARK
---metadata cuda-version=${CUDA_VERSION}
 --metadata benchmark-home=${BENCHMARK_HOME}
 EOF
 )