Update build.sh (#63200)

PaddlePaddle · Apr 3, 2024 · 5f32f92 · 5f32f92
1 parent 401bfb1
commit 5f32f92
Showing 1 changed file with 6 additions and 6 deletions.
diff --git 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)