diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh index bcecab8ba356fe854f9dc03ac608e3f478c66307..e8af7c5a3625b90ae2a00433fd833bb8f44ab8d1 100755 --- a/paddle/scripts/paddle_build.sh +++ b/paddle/scripts/paddle_build.sh @@ -262,6 +262,7 @@ function cmake_base() { -DWITH_RECORD_BUILDTIME=${WITH_RECORD_BUILDTIME:-OFF} -DCUDA_ARCH_BIN="${CUDA_ARCH_BIN}" -DWITH_ONNXRUNTIME=${WITH_ONNXRUNTIME:-OFF} + -DWITH_CUDNN_FRONTEND=${WITH_CUDNN_FRONTEND:-OFF} ======================================== EOF # Disable UNITTEST_USE_VIRTUALENV in docker because @@ -318,7 +319,8 @@ EOF -DCUDA_ARCH_BIN="${CUDA_ARCH_BIN}" \ -DWITH_RECORD_BUILDTIME=${WITH_RECORD_BUILDTIME:-OFF} \ -DWITH_UNITY_BUILD=${WITH_UNITY_BUILD:-OFF} \ - -DWITH_ONNXRUNTIME=${WITH_ONNXRUNTIME:-OFF};build_error=$? + -DWITH_ONNXRUNTIME=${WITH_ONNXRUNTIME:-OFF} \ + -DWITH_CUDNN_FRONTEND=${WITH_CUDNN_FRONTEND:-OFF};build_error=$? if [ "$build_error" != 0 ];then exit 7;