From 5f32f92640792024d47174ae22db1b8a07a3c512 Mon Sep 17 00:00:00 2001 From: co63oc Date: Wed, 3 Apr 2024 17:51:07 +0800 Subject: [PATCH] Update build.sh (#63200) --- tools/cinn/build.sh | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tools/cinn/build.sh b/tools/cinn/build.sh index 600cfd687c75a..ca2662ebe4675 100755 --- a/tools/cinn/build.sh +++ b/tools/cinn/build.sh @@ -32,13 +32,13 @@ cuda_config=OFF cudnn_config=OFF mklcblas_config=ON -mkldnn_config=ON +onednn_config=ON function mklcblas_off { mklcblas_config=OFF } -function mkldnn_off { - mkldnn_config=OFF +function onednn_off { + onednn_config=OFF } @@ -204,11 +204,11 @@ function main { case $i in mklcblas_off) mklcblas_off - mkldnn_off + onednn_off shift ;; - mkldnn_off) - mkldnn_off + onednn_off) + onednn_off shift ;; gpu_on)