From 12fc18c22f6b4f5818d07d77e12f7a90fa0417fd Mon Sep 17 00:00:00 2001
From: xionglei <xionglei@baidu.com>
Date: Thu, 9 Nov 2017 20:47:54 +0800
Subject: [PATCH] add API for copying data from/to paddle matrix
---
paddle/capi/Matrix.cpp | 40 ++++++++++++++++++++++++++++++++++++++++
paddle/capi/matrix.h | 19 +++++++++++++++++++
2 files changed, 59 insertions(+)
diff --git a/paddle/capi/Matrix.cpp b/paddle/capi/Matrix.cpp
index 4547afaf1d..53a36f8f20 100644
--- a/paddle/capi/Matrix.cpp
+++ b/paddle/capi/Matrix.cpp
@@ -54,6 +54,46 @@ paddle_error paddle_matrix_set_row(paddle_matrix mat,
return kPD_NO_ERROR;
}
+PD_API paddle_error paddle_matrix_set_value(paddle_matrix mat,
+ paddle_real* value) {
+ if (mat == nullptr || value == nullptr) return kPD_NULLPTR;
+ auto ptr = cast(mat);
+ if (ptr->mat == nullptr) return kPD_NULLPTR;
+ paddle::real* buf = ptr->mat->getRowBuf(0);
+ size_t width = ptr->mat->getWidth();
+ size_t height = ptr->mat->getHeight();
+ if (ptr->mat->useGpu()) {
+#ifdef PADDLE_WITH_CUDA
+ hl_memcpy(buf, value, sizeof(paddle::real) * width * height);
+#else
+ return kPD_NOT_SUPPORTED;
+#endif
+ } else {
+ std::copy(value, value + width * height, buf);
+ }
+ return kPD_NO_ERROR;
+}
+
+PD_API paddle_error paddle_matrix_get_value(paddle_matrix mat,
+ paddle_real* result) {
+ if (mat == nullptr || result == nullptr) return kPD_NULLPTR;
+ auto ptr = cast(mat);
+ if (ptr->mat == nullptr) return kPD_NULLPTR;
+ paddle::real* buf = ptr->mat->getRowBuf(0);
+ size_t width = ptr->mat->getWidth();
+ size_t height = ptr->mat->getHeight();
+ if (ptr->mat->useGpu()) {
+#ifdef PADDLE_WITH_CUDA
+ hl_memcpy(result, buf, width * height * sizeof(paddle::real));
+#else
+ return kPD_NOT_SUPPORTED;
+#endif
+ } else {
+ std::copy(buf, buf + width * height, result);
+ }
+ return kPD_NO_ERROR;
+}
+
paddle_error paddle_matrix_get_row(paddle_matrix mat,
uint64_t rowID,
paddle_real** rawRowBuffer) {
diff --git a/paddle/capi/matrix.h b/paddle/capi/matrix.h
index f15f7f3bbb..bb5223f8a2 100644
--- a/paddle/capi/matrix.h
+++ b/paddle/capi/matrix.h
@@ -70,6 +70,16 @@ PD_API paddle_error paddle_matrix_set_row(paddle_matrix mat,
uint64_t rowID,
paddle_real* rowArray);
+/**
+ * @brief paddle_matrix_set_value Set value to matrix.
+ * @param mat Target Matrix
+ * @param value Row data.
+ * @return paddle_error
+ * @note value should contain enough element of data to init the mat
+ */
+PD_API paddle_error paddle_matrix_set_value(paddle_matrix mat,
+ paddle_real* value);
+
/**
* @brief PDMatGetRow Get raw row buffer from matrix
* @param [in] mat Target matrix
@@ -81,6 +91,15 @@ PD_API paddle_error paddle_matrix_get_row(paddle_matrix mat,
uint64_t rowID,
paddle_real** rawRowBuffer);
+/**
+ * @brief copy data from the matrix
+ * @param [in] mat Target matrix
+ * @param [out] result pointer to store the matrix data
+ * @return paddle_error
+ * @note the space of the result should allocated before invoke this API
+ */
+PD_API paddle_error paddle_matrix_get_value(paddle_matrix mat,
+ paddle_real* result);
/**
* @brief PDMatCreateNone Create None Matrix
* @return
--
GitLab