修复 infinicore.Tensor.data_ptr 的调用问题

Author: voltjia

python/infinicore/tensor.py

@@ -32,7 +32,7 @@ def ndim(self):
         return self._underlying.ndim
 
     def data_ptr(self):
-        return self._underlying.data_ptr
+        return self._underlying.data_ptr()
 
     def size(self, dim=None):
         if dim is None:

src/infinicore/pybind11/tensor.hpp

@@ -17,7 +17,7 @@ inline void bind(py::module &m) {
         .def_property_readonly("dtype", [](const Tensor &tensor) { return tensor->dtype(); })
         .def_property_readonly("device", [](const Tensor &tensor) { return tensor->device(); })
 
-        .def("data_ptr", [](const Tensor &tensor) { return tensor->data(); })
+        .def("data_ptr", [](const Tensor &tensor) { return reinterpret_cast<std::uintptr_t>(tensor->data()); })
         .def("size", [](const Tensor &tensor, std::size_t dim) { return tensor->size(dim); })
         .def("stride", [](const Tensor &tensor, std::size_t dim) { return tensor->stride(dim); })
         .def("numel", [](const Tensor &tensor) { return tensor->numel(); })