add varbase_copy support CUDAPinnedPlace (#32883)

48fc16f2 · chentianyu03 · GitHub · c3ae0d40 · 48fc16f2
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/pybind/imperative.cc paddle/fluid/pybind/imperative.cc +1 -0

未找到文件。
--- 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",