From f0cd5142afbde12182ba532fbbb89c3b4e491f34 Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Fri, 22 Sep 2017 11:13:42 -0700 Subject: [PATCH] Change namespace of pybind.cc to pybind --- paddle/framework/tensor.h | 7 +++++-- paddle/pybind/pybind.cc | 15 +++++++-------- paddle/pybind/tensor_py.h | 2 +- 3 files changed, 13 insertions(+), 11 deletions(-) diff --git a/paddle/framework/tensor.h b/paddle/framework/tensor.h index 4b5a2ae523f..6a14cfde4b7 100644 --- a/paddle/framework/tensor.h +++ b/paddle/framework/tensor.h @@ -29,16 +29,19 @@ limitations under the License. */ namespace paddle { -namespace framework { +namespace pybind { namespace details { template struct CastToPyBufferImpl; } +} // namespace pybind + +namespace framework { class Tensor { public: template - friend struct details::CastToPyBufferImpl; + friend struct pybind::details::CastToPyBufferImpl; template friend struct EigenTensor; diff --git a/paddle/pybind/pybind.cc b/paddle/pybind/pybind.cc index c7009a604f6..9f5451340d5 100644 --- a/paddle/pybind/pybind.cc +++ b/paddle/pybind/pybind.cc @@ -34,12 +34,7 @@ limitations under the License. */ namespace py = pybind11; namespace paddle { -namespace framework { - -using Tensor = framework::Tensor; -using LoDTensor = framework::LoDTensor; -using LoD = framework::LoD; - +namespace pybind { static size_t UniqueIntegerGenerator() { static std::atomic generator; return generator.fetch_add(1); @@ -56,6 +51,10 @@ bool IsCompileGPU() { PYBIND11_PLUGIN(core) { py::module m("core", "C++ core of PaddlePaddle"); + // using framework in this function. Since it is inside a function, it will + // not cause namespace pollution. + using namespace paddle::framework; // NOLINT + py::class_(m, "Tensor", py::buffer_protocol()) .def_buffer( [](Tensor &self) -> py::buffer_info { return CastToPyBuffer(self); }) @@ -63,7 +62,7 @@ PYBIND11_PLUGIN(core) { [](const Tensor &self) { return vectorize(self.dims()); }) .def("set_dims", [](Tensor &self, const std::vector &dim) { - self.Resize(make_ddim(dim)); + self.Resize(framework::make_ddim(dim)); }) .def("alloc_float", [](Tensor &self, paddle::platform::GPUPlace &place) { @@ -317,5 +316,5 @@ All parameter, weight, gradient are variables in Paddle. return m.ptr(); } -} // namespace framework +} // namespace pybind } // namespace paddle diff --git a/paddle/pybind/tensor_py.h b/paddle/pybind/tensor_py.h index 95171acf729..bcfba84a1aa 100644 --- a/paddle/pybind/tensor_py.h +++ b/paddle/pybind/tensor_py.h @@ -23,7 +23,7 @@ namespace py = pybind11; namespace paddle { -namespace framework { +namespace pybind { namespace details { -- GitLab