From 64f52574f8a0097db7e365e1de458c567925d669 Mon Sep 17 00:00:00 2001
From: "huheng.1989" <huheng.1989@bytedance.com>
Date: Sun, 18 Feb 2024 11:26:02 +0800
Subject: [PATCH] build.sh: use disable_cuda option

---
 build.sh              | 8 ++------
 docker/Dockerfile.cpu | 2 +-
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/build.sh b/build.sh
index 2fd0cb29..c831231f 100755
--- a/build.sh
+++ b/build.sh
@@ -60,13 +60,9 @@ then
     then
         LOCAL_BUILD=0
     fi
-    if [ "$1" = "disable" ]
+    if [ "$1" = "disable_cuda" ]
     then
-        shift
-        if [ "$1" = "cuda" ]
-        then
-            CUDA_ENABLE=OFF
-        fi
+        CUDA_ENABLE=OFF
     fi
 fi
 
diff --git a/docker/Dockerfile.cpu b/docker/Dockerfile.cpu
index 26dc6e18..c243f551 100644
--- a/docker/Dockerfile.cpu
+++ b/docker/Dockerfile.cpu
@@ -22,7 +22,7 @@ WORKDIR /root/bmf/
 RUN scripts/build_ffmpeg.sh --device=$DEVICE nasm yasm x264 x265 fdk-aac opus \
     && rm -rf ffmpeg_source
 
-RUN ./build.sh disable cuda && rm -rf .git && ./build.sh clean
+RUN ./build.sh disable_cuda && rm -rf .git && ./build.sh clean
 
 ENV LD_LIBRARY_PATH=/root/bmf/output/bmf/lib
 ENV PATH=${PATH}:/root/bmf/output/bmf/bin