fix compile error

PaddlePaddle · Mar 18, 2023 · 115cfb1 · 115cfb1
1 parent 52d3b62
commit 115cfb1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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 <cuda_runtime_api.h>
 #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"