From 605bc003c510b2db3f3c3ae8535ea2302fbe553f Mon Sep 17 00:00:00 2001 From: Tian Zheng Date: Thu, 3 Nov 2022 16:31:19 +0800 Subject: [PATCH] Test CUDNN Frontend Build (#47612) --- paddle/scripts/paddle_build.sh | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/paddle/scripts/paddle_build.sh b/paddle/scripts/paddle_build.sh index bcecab8ba3..e8af7c5a36 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; -- GitLab