From 48fc16f2c2f0102c7017cf2b8a210d0abf1ab68e Mon Sep 17 00:00:00 2001
From: chentianyu03 <chentianyu03@baidu.com>
Date: Thu, 13 May 2021 14:38:54 +0800
Subject: [PATCH] add varbase_copy support CUDAPinnedPlace (#32883)

---
 paddle/fluid/pybind/imperative.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/fluid/pybind/imperative.cc b/paddle/fluid/pybind/imperative.cc
index 4bdf1e21bac..825b837a732 100644
--- a/paddle/fluid/pybind/imperative.cc
+++ b/paddle/fluid/pybind/imperative.cc
@@ -1699,6 +1699,7 @@ void BindImperative(py::module *m_ptr) {
   m.def("varbase_copy", &VarBaseCopy<platform::CPUPlace>);
   m.def("varbase_copy", &VarBaseCopy<platform::CUDAPlace>);
   m.def("varbase_copy", &VarBaseCopy<platform::XPUPlace>);
+  m.def("varbase_copy", &VarBaseCopy<platform::CUDAPinnedPlace>);
 
   m.def(
       "dygraph_partial_grad",
-- 
GitLab