From 115cfb13705e6f6d7a6d041f0f36499ef01568eb Mon Sep 17 00:00:00 2001 From: sneaxiy Date: Sat, 18 Mar 2023 08:49:34 +0800 Subject: [PATCH] fix compile error --- paddle/phi/kernels/funcs/blas/blaslt_impl.cu.h | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/phi/kernels/funcs/blas/blaslt_impl.cu.h b/paddle/phi/kernels/funcs/blas/blaslt_impl.cu.h index 755c98b2a5fc6..5262fc1210c01 100644 --- a/paddle/phi/kernels/funcs/blas/blaslt_impl.cu.h +++ b/paddle/phi/kernels/funcs/blas/blaslt_impl.cu.h @@ -18,6 +18,8 @@ limitations under the License. */ #include #include "cuda.h" // NOLINT +#include "paddle/fluid/memory/malloc.h" +#include "paddle/phi/backends/dynload/cublasLt.h" #include "paddle/phi/backends/gpu/cuda/cuda_helper.h" #include "paddle/phi/common/amp_type_traits.h" #include "paddle/phi/kernels/autotune/cache.h"