From a404d9abb4e57b9c903ad98becca45b43d7d92d6 Mon Sep 17 00:00:00 2001 From: Yu Yang Date: Wed, 2 Aug 2017 17:43:17 +0800 Subject: [PATCH] Add todo comments --- paddle/pybind/pybind.cc | 2 ++ 1 file changed, 2 insertions(+) diff --git a/paddle/pybind/pybind.cc b/paddle/pybind/pybind.cc index fa76da67f26..dc6f29d026b 100644 --- a/paddle/pybind/pybind.cc +++ b/paddle/pybind/pybind.cc @@ -81,9 +81,11 @@ PYBIND11_PLUGIN(core) { .def("shape", [](pd::Tensor& self) { return pd::vectorize(self.dims()); }) .def("set_float_element", [](pd::Tensor& self, size_t offset, float f) { + // TODO(yuyang18): Only support GPU now. self.data()[offset] = f; }) .def("get_float_element", [](pd::Tensor& self, size_t offset) -> float { + // TODO(yuyang18): Only support GPU now. return self.data()[offset]; }); -- GitLab