Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
Paddle
提交
66a28e13
P
Paddle
项目概览
PaddlePaddle
/
Paddle
1 年多 前同步成功
通知
2302
Star
20931
Fork
5422
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1423
列表
看板
标记
里程碑
合并请求
543
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1,423
Issue
1,423
列表
看板
标记
里程碑
合并请求
543
合并请求
543
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
未验证
提交
66a28e13
编写于
6月 23, 2022
作者:
R
Ruibiao Chen
提交者:
GitHub
6月 23, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Remove unnecessary includings for pstring.h (#43752)
* Remove unnecessary including for pstring.h * Fix typos
上级
e3d94fc5
变更
7
隐藏空白更改
内联
并排
Showing
7 changed file
with
174 addition
and
88 deletion
+174
-88
paddle/fluid/framework/convert_utils.cc
paddle/fluid/framework/convert_utils.cc
+2
-0
paddle/fluid/pybind/tensor_py.h
paddle/fluid/pybind/tensor_py.h
+159
-86
paddle/phi/common/data_type.h
paddle/phi/common/data_type.h
+6
-1
paddle/phi/core/string_tensor.cc
paddle/phi/core/string_tensor.cc
+1
-0
paddle/phi/core/string_tensor.h
paddle/phi/core/string_tensor.h
+4
-1
paddle/phi/kernels/strings/cpu/strings_copy_kernel.cc
paddle/phi/kernels/strings/cpu/strings_copy_kernel.cc
+1
-0
paddle/phi/tests/api/test_strings_lower_upper_api.cc
paddle/phi/tests/api/test_strings_lower_upper_api.cc
+1
-0
未找到文件。
paddle/fluid/framework/convert_utils.cc
浏览文件 @
66a28e13
...
@@ -11,10 +11,12 @@ distributed under the License is distributed on an "AS IS" BASIS,
...
@@ -11,10 +11,12 @@ distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
See the License for the specific language governing permissions and
limitations under the License. */
limitations under the License. */
#include "paddle/fluid/framework/convert_utils.h"
#include "paddle/fluid/framework/convert_utils.h"
// See Note [ Why still include the fluid headers? ]
// See Note [ Why still include the fluid headers? ]
#include "paddle/fluid/platform/device/gpu/gpu_info.h"
#include "paddle/fluid/platform/device/gpu/gpu_info.h"
#include "paddle/phi/common/pstring.h"
namespace
paddle
{
namespace
paddle
{
namespace
framework
{
namespace
framework
{
...
...
paddle/fluid/pybind/tensor_py.h
浏览文件 @
66a28e13
...
@@ -38,6 +38,7 @@ limitations under the License. */
...
@@ -38,6 +38,7 @@ limitations under the License. */
#include "paddle/fluid/platform/device_context.h"
#include "paddle/fluid/platform/device_context.h"
#include "paddle/fluid/platform/float16.h"
#include "paddle/fluid/platform/float16.h"
#include "paddle/fluid/platform/profiler/event_tracing.h"
#include "paddle/fluid/platform/profiler/event_tracing.h"
#include "paddle/phi/common/pstring.h"
#include "paddle/phi/core/string_tensor.h"
#include "paddle/phi/core/string_tensor.h"
#include "paddle/phi/kernels/strings/unicode.h"
#include "paddle/phi/kernels/strings/unicode.h"
#include "pybind11/numpy.h"
#include "pybind11/numpy.h"
...
@@ -180,14 +181,17 @@ template <typename T>
...
@@ -180,14 +181,17 @@ template <typename T>
class
PYBIND11_HIDDEN
NumpyAllocation
:
public
memory
::
Allocation
{
class
PYBIND11_HIDDEN
NumpyAllocation
:
public
memory
::
Allocation
{
public:
public:
explicit
NumpyAllocation
(
const
py
::
array
&
arr
)
explicit
NumpyAllocation
(
const
py
::
array
&
arr
)
:
Allocation
(
const_cast
<
void
*>
(
arr
.
data
()),
sizeof
(
T
)
*
(
arr
.
size
()),
:
Allocation
(
const_cast
<
void
*>
(
arr
.
data
()),
sizeof
(
T
)
*
(
arr
.
size
()),
paddle
::
platform
::
CPUPlace
()),
paddle
::
platform
::
CPUPlace
()),
arr_
(
arr
.
ptr
())
{
arr_
(
arr
.
ptr
())
{
PADDLE_ENFORCE_NOT_NULL
(
arr_
,
platform
::
errors
::
InvalidArgument
(
PADDLE_ENFORCE_NOT_NULL
(
"The underlying PyObject pointer of "
arr_
,
"numpy array cannot be nullptr"
));
platform
::
errors
::
InvalidArgument
(
"The underlying PyObject pointer of "
"numpy array cannot be nullptr"
));
PADDLE_ENFORCE_NE
(
PADDLE_ENFORCE_NE
(
arr_
,
Py_None
,
arr_
,
Py_None
,
platform
::
errors
::
PreconditionNotMet
(
platform
::
errors
::
PreconditionNotMet
(
"The underlying PyObject pointer of numpy array cannot be None"
));
"The underlying PyObject pointer of numpy array cannot be None"
));
Py_INCREF
(
arr_
);
Py_INCREF
(
arr_
);
...
@@ -241,7 +245,8 @@ inline std::string TensorDTypeToPyDTypeStr(
...
@@ -241,7 +245,8 @@ inline std::string TensorDTypeToPyDTypeStr(
} else { \
} else { \
constexpr auto kIsValidDType = ValidDTypeToPyArrayChecker<T>::kValue; \
constexpr auto kIsValidDType = ValidDTypeToPyArrayChecker<T>::kValue; \
PADDLE_ENFORCE_EQ( \
PADDLE_ENFORCE_EQ( \
kIsValidDType, true, \
kIsValidDType, \
true, \
platform::errors::Unimplemented( \
platform::errors::Unimplemented( \
"This type [%s] of tensor cannot be expose to Python", \
"This type [%s] of tensor cannot be expose to Python", \
typeid(T).name())); \
typeid(T).name())); \
...
@@ -259,7 +264,8 @@ inline std::string TensorDTypeToPyDTypeStr(
...
@@ -259,7 +264,8 @@ inline std::string TensorDTypeToPyDTypeStr(
template
<
typename
T
>
template
<
typename
T
>
T
TensorGetElement
(
const
framework
::
Tensor
&
self
,
size_t
offset
)
{
T
TensorGetElement
(
const
framework
::
Tensor
&
self
,
size_t
offset
)
{
PADDLE_ENFORCE_LT
(
offset
,
self
.
numel
(),
PADDLE_ENFORCE_LT
(
offset
,
self
.
numel
(),
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"The offset exceeds the size of tensor."
));
"The offset exceeds the size of tensor."
));
...
@@ -276,29 +282,29 @@ T TensorGetElement(const framework::Tensor &self, size_t offset) {
...
@@ -276,29 +282,29 @@ T TensorGetElement(const framework::Tensor &self, size_t offset) {
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
const
T
*
a
=
self
.
data
<
T
>
();
const
T
*
a
=
self
.
data
<
T
>
();
auto
p
=
self
.
place
();
auto
p
=
self
.
place
();
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
&
b
,
p
,
a
+
offset
,
sizeof
(
T
),
paddle
::
memory
::
Copy
(
nullptr
);
platform
::
CPUPlace
(),
&
b
,
p
,
a
+
offset
,
sizeof
(
T
),
nullptr
);
#endif
#endif
}
else
if
(
platform
::
is_mlu_place
(
self
.
place
()))
{
}
else
if
(
platform
::
is_mlu_place
(
self
.
place
()))
{
#ifdef PADDLE_WITH_MLU
#ifdef PADDLE_WITH_MLU
const
T
*
a
=
self
.
data
<
T
>
();
const
T
*
a
=
self
.
data
<
T
>
();
auto
p
=
self
.
place
();
auto
p
=
self
.
place
();
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
&
b
,
p
,
a
+
offset
,
sizeof
(
T
),
paddle
::
memory
::
Copy
(
nullptr
);
platform
::
CPUPlace
(),
&
b
,
p
,
a
+
offset
,
sizeof
(
T
),
nullptr
);
#endif
#endif
}
else
if
(
platform
::
is_npu_place
(
self
.
place
()))
{
}
else
if
(
platform
::
is_npu_place
(
self
.
place
()))
{
#if defined(PADDLE_WITH_ASCEND_CL)
#if defined(PADDLE_WITH_ASCEND_CL)
const
T
*
a
=
self
.
data
<
T
>
();
const
T
*
a
=
self
.
data
<
T
>
();
auto
p
=
self
.
place
();
auto
p
=
self
.
place
();
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
&
b
,
p
,
a
+
offset
,
sizeof
(
T
),
paddle
::
memory
::
Copy
(
nullptr
);
platform
::
CPUPlace
(),
&
b
,
p
,
a
+
offset
,
sizeof
(
T
),
nullptr
);
#endif
#endif
}
else
if
(
platform
::
is_custom_place
(
self
.
place
()))
{
}
else
if
(
platform
::
is_custom_place
(
self
.
place
()))
{
#if defined(PADDLE_WITH_CUSTOM_DEVICE)
#if defined(PADDLE_WITH_CUSTOM_DEVICE)
const
T
*
a
=
self
.
data
<
T
>
();
const
T
*
a
=
self
.
data
<
T
>
();
auto
p
=
self
.
place
();
auto
p
=
self
.
place
();
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
&
b
,
p
,
a
+
offset
,
sizeof
(
T
),
paddle
::
memory
::
Copy
(
nullptr
);
platform
::
CPUPlace
(),
&
b
,
p
,
a
+
offset
,
sizeof
(
T
),
nullptr
);
#endif
#endif
}
}
VLOG
(
10
)
<<
"TensorGetElement, place: "
<<
self
.
place
()
VLOG
(
10
)
<<
"TensorGetElement, place: "
<<
self
.
place
()
...
@@ -308,7 +314,8 @@ T TensorGetElement(const framework::Tensor &self, size_t offset) {
...
@@ -308,7 +314,8 @@ T TensorGetElement(const framework::Tensor &self, size_t offset) {
template
<
typename
T
>
template
<
typename
T
>
void
TensorSetElement
(
framework
::
Tensor
*
self
,
size_t
offset
,
T
elem
)
{
void
TensorSetElement
(
framework
::
Tensor
*
self
,
size_t
offset
,
T
elem
)
{
PADDLE_ENFORCE_LT
(
offset
,
self
->
numel
(),
PADDLE_ENFORCE_LT
(
offset
,
self
->
numel
(),
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"The offset exceeds the size of tensor."
));
"The offset exceeds the size of tensor."
));
VLOG
(
10
)
<<
"TensorSetElement, place: "
<<
self
->
place
()
VLOG
(
10
)
<<
"TensorSetElement, place: "
<<
self
->
place
()
...
@@ -325,29 +332,29 @@ void TensorSetElement(framework::Tensor *self, size_t offset, T elem) {
...
@@ -325,29 +332,29 @@ void TensorSetElement(framework::Tensor *self, size_t offset, T elem) {
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
auto
p
=
self
->
place
();
auto
p
=
self
->
place
();
T
*
a
=
self
->
mutable_data
<
T
>
(
p
);
T
*
a
=
self
->
mutable_data
<
T
>
(
p
);
paddle
::
memory
::
Copy
(
p
,
a
+
offset
,
platform
::
CPUPlace
(),
&
elem
,
sizeof
(
T
),
paddle
::
memory
::
Copy
(
nullptr
);
p
,
a
+
offset
,
platform
::
CPUPlace
(),
&
elem
,
sizeof
(
T
),
nullptr
);
#endif
#endif
}
else
if
(
platform
::
is_mlu_place
(
self
->
place
()))
{
}
else
if
(
platform
::
is_mlu_place
(
self
->
place
()))
{
#ifdef PADDLE_WITH_MLU
#ifdef PADDLE_WITH_MLU
auto
p
=
self
->
place
();
auto
p
=
self
->
place
();
T
*
a
=
self
->
mutable_data
<
T
>
(
p
);
T
*
a
=
self
->
mutable_data
<
T
>
(
p
);
paddle
::
memory
::
Copy
(
p
,
a
+
offset
,
platform
::
CPUPlace
(),
&
elem
,
sizeof
(
T
),
paddle
::
memory
::
Copy
(
nullptr
);
p
,
a
+
offset
,
platform
::
CPUPlace
(),
&
elem
,
sizeof
(
T
),
nullptr
);
#endif
#endif
}
else
if
(
platform
::
is_npu_place
(
self
->
place
()))
{
}
else
if
(
platform
::
is_npu_place
(
self
->
place
()))
{
#if defined(PADDLE_WITH_ASCEND_CL)
#if defined(PADDLE_WITH_ASCEND_CL)
auto
p
=
self
->
place
();
auto
p
=
self
->
place
();
T
*
a
=
self
->
mutable_data
<
T
>
(
p
);
T
*
a
=
self
->
mutable_data
<
T
>
(
p
);
paddle
::
memory
::
Copy
(
p
,
a
+
offset
,
platform
::
CPUPlace
(),
&
elem
,
sizeof
(
T
),
paddle
::
memory
::
Copy
(
nullptr
);
p
,
a
+
offset
,
platform
::
CPUPlace
(),
&
elem
,
sizeof
(
T
),
nullptr
);
#endif
#endif
}
else
if
(
platform
::
is_custom_place
(
self
->
place
()))
{
}
else
if
(
platform
::
is_custom_place
(
self
->
place
()))
{
#if defined(PADDLE_WITH_CUSTOM_DEVICE)
#if defined(PADDLE_WITH_CUSTOM_DEVICE)
auto
p
=
self
->
place
();
auto
p
=
self
->
place
();
T
*
a
=
self
->
mutable_data
<
T
>
(
p
);
T
*
a
=
self
->
mutable_data
<
T
>
(
p
);
paddle
::
memory
::
Copy
(
p
,
a
+
offset
,
platform
::
CPUPlace
(),
&
elem
,
sizeof
(
T
),
paddle
::
memory
::
Copy
(
nullptr
);
p
,
a
+
offset
,
platform
::
CPUPlace
(),
&
elem
,
sizeof
(
T
),
nullptr
);
#endif
#endif
}
}
}
}
...
@@ -356,7 +363,8 @@ template <typename T, typename P>
...
@@ -356,7 +363,8 @@ template <typename T, typename P>
void
SetTensorFromPyArrayT
(
void
SetTensorFromPyArrayT
(
framework
::
Tensor
*
self
,
framework
::
Tensor
*
self
,
const
py
::
array_t
<
T
,
py
::
array
::
c_style
|
py
::
array
::
forcecast
>
&
array
,
const
py
::
array_t
<
T
,
py
::
array
::
c_style
|
py
::
array
::
forcecast
>
&
array
,
const
P
&
place
,
bool
zero_copy
)
{
const
P
&
place
,
bool
zero_copy
)
{
std
::
vector
<
int64_t
>
dims
;
std
::
vector
<
int64_t
>
dims
;
dims
.
reserve
(
array
.
ndim
());
dims
.
reserve
(
array
.
ndim
());
for
(
decltype
(
array
.
ndim
())
i
=
0
;
i
<
array
.
ndim
();
++
i
)
{
for
(
decltype
(
array
.
ndim
())
i
=
0
;
i
<
array
.
ndim
();
++
i
)
{
...
@@ -380,8 +388,11 @@ void SetTensorFromPyArrayT(
...
@@ -380,8 +388,11 @@ void SetTensorFromPyArrayT(
platform
::
Place
tmp_place
=
place
;
platform
::
Place
tmp_place
=
place
;
platform
::
XPUDeviceGuard
guard
(
tmp_place
.
device
);
platform
::
XPUDeviceGuard
guard
(
tmp_place
.
device
);
auto
dst
=
self
->
mutable_data
<
T
>
(
place
);
auto
dst
=
self
->
mutable_data
<
T
>
(
place
);
memory
::
Copy
(
tmp_place
,
static_cast
<
void
*>
(
dst
),
platform
::
CPUPlace
(),
memory
::
Copy
(
tmp_place
,
static_cast
<
const
void
*>
(
array
.
data
()),
array
.
nbytes
());
static_cast
<
void
*>
(
dst
),
platform
::
CPUPlace
(),
static_cast
<
const
void
*>
(
array
.
data
()),
array
.
nbytes
());
#else
#else
PADDLE_THROW
(
platform
::
errors
::
PermissionDenied
(
PADDLE_THROW
(
platform
::
errors
::
PermissionDenied
(
"Cannot use XPUPlace in CPU/GPU version, "
"Cannot use XPUPlace in CPU/GPU version, "
...
@@ -413,8 +424,8 @@ void SetTensorFromPyArrayT(
...
@@ -413,8 +424,8 @@ void SetTensorFromPyArrayT(
platform
::
Place
tmp_place
=
place
;
platform
::
Place
tmp_place
=
place
;
platform
::
NPUDeviceGuard
guard
(
tmp_place
.
device
);
platform
::
NPUDeviceGuard
guard
(
tmp_place
.
device
);
auto
dst
=
self
->
mutable_data
<
T
>
(
place
);
auto
dst
=
self
->
mutable_data
<
T
>
(
place
);
platform
::
NPUMemcpySync
(
dst
,
array
.
data
(),
array
.
nbytes
(),
platform
::
NPUMemcpySync
(
ACL_MEMCPY_HOST_TO_DEVICE
);
dst
,
array
.
data
(),
array
.
nbytes
(),
ACL_MEMCPY_HOST_TO_DEVICE
);
platform
::
DeviceContextPool
&
pool
=
platform
::
DeviceContextPool
::
Instance
();
platform
::
DeviceContextPool
&
pool
=
platform
::
DeviceContextPool
::
Instance
();
auto
&
ctx
=
*
pool
.
Get
(
place
);
auto
&
ctx
=
*
pool
.
Get
(
place
);
ctx
.
Wait
();
ctx
.
Wait
();
...
@@ -460,11 +471,11 @@ void SetTensorFromPyArrayT(
...
@@ -460,11 +471,11 @@ void SetTensorFromPyArrayT(
platform
::
CUDADeviceGuard
guard
(
place
.
device
);
platform
::
CUDADeviceGuard
guard
(
place
.
device
);
auto
dst
=
self
->
mutable_data
<
T
>
(
place
);
auto
dst
=
self
->
mutable_data
<
T
>
(
place
);
#ifdef PADDLE_WITH_HIP
#ifdef PADDLE_WITH_HIP
paddle
::
platform
::
GpuMemcpySync
(
dst
,
array
.
data
(),
array
.
nbytes
(),
paddle
::
platform
::
GpuMemcpySync
(
hipMemcpyHostToDevice
);
dst
,
array
.
data
(),
array
.
nbytes
(),
hipMemcpyHostToDevice
);
#else
#else
paddle
::
platform
::
GpuMemcpySync
(
dst
,
array
.
data
(),
array
.
nbytes
(),
paddle
::
platform
::
GpuMemcpySync
(
cudaMemcpyHostToDevice
);
dst
,
array
.
data
(),
array
.
nbytes
(),
cudaMemcpyHostToDevice
);
#endif
#endif
}
else
if
(
paddle
::
platform
::
is_cuda_pinned_place
(
place
))
{
}
else
if
(
paddle
::
platform
::
is_cuda_pinned_place
(
place
))
{
...
@@ -486,8 +497,10 @@ void SetTensorFromPyArrayT(
...
@@ -486,8 +497,10 @@ void SetTensorFromPyArrayT(
}
}
template
<
typename
P
>
template
<
typename
P
>
void
SetTensorFromPyArray
(
framework
::
Tensor
*
self
,
const
py
::
object
&
obj
,
void
SetTensorFromPyArray
(
framework
::
Tensor
*
self
,
const
P
&
place
,
bool
zero_copy
)
{
const
py
::
object
&
obj
,
const
P
&
place
,
bool
zero_copy
)
{
auto
array
=
obj
.
cast
<
py
::
array
>
();
auto
array
=
obj
.
cast
<
py
::
array
>
();
if
(
py
::
isinstance
<
py
::
array_t
<
float
>>
(
array
))
{
if
(
py
::
isinstance
<
py
::
array_t
<
float
>>
(
array
))
{
SetTensorFromPyArrayT
<
float
,
P
>
(
self
,
array
,
place
,
zero_copy
);
SetTensorFromPyArrayT
<
float
,
P
>
(
self
,
array
,
place
,
zero_copy
);
...
@@ -504,8 +517,8 @@ void SetTensorFromPyArray(framework::Tensor *self, const py::object &obj,
...
@@ -504,8 +517,8 @@ void SetTensorFromPyArray(framework::Tensor *self, const py::object &obj,
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
uint8_t
>>
(
array
))
{
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
uint8_t
>>
(
array
))
{
SetTensorFromPyArrayT
<
uint8_t
,
P
>
(
self
,
array
,
place
,
zero_copy
);
SetTensorFromPyArrayT
<
uint8_t
,
P
>
(
self
,
array
,
place
,
zero_copy
);
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
paddle
::
platform
::
float16
>>
(
array
))
{
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
paddle
::
platform
::
float16
>>
(
array
))
{
SetTensorFromPyArrayT
<
paddle
::
platform
::
float16
,
P
>
(
self
,
array
,
place
,
SetTensorFromPyArrayT
<
paddle
::
platform
::
float16
,
P
>
(
zero_copy
);
self
,
array
,
place
,
zero_copy
);
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
paddle
::
platform
::
complex
<
float
>>>
(
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
paddle
::
platform
::
complex
<
float
>>>
(
array
))
{
array
))
{
SetTensorFromPyArrayT
<
paddle
::
platform
::
complex
<
float
>
,
P
>
(
SetTensorFromPyArrayT
<
paddle
::
platform
::
complex
<
float
>
,
P
>
(
...
@@ -517,8 +530,8 @@ void SetTensorFromPyArray(framework::Tensor *self, const py::object &obj,
...
@@ -517,8 +530,8 @@ void SetTensorFromPyArray(framework::Tensor *self, const py::object &obj,
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
uint16_t
>>
(
array
))
{
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
uint16_t
>>
(
array
))
{
// since there is still no support for bfloat16 in NumPy,
// since there is still no support for bfloat16 in NumPy,
// uint16 is used for casting bfloat16
// uint16 is used for casting bfloat16
SetTensorFromPyArrayT
<
paddle
::
platform
::
bfloat16
,
P
>
(
self
,
array
,
place
,
SetTensorFromPyArrayT
<
paddle
::
platform
::
bfloat16
,
P
>
(
zero_copy
);
self
,
array
,
place
,
zero_copy
);
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
bool
>>
(
array
))
{
}
else
if
(
py
::
isinstance
<
py
::
array_t
<
bool
>>
(
array
))
{
SetTensorFromPyArrayT
<
bool
,
P
>
(
self
,
array
,
place
,
zero_copy
);
SetTensorFromPyArrayT
<
bool
,
P
>
(
self
,
array
,
place
,
zero_copy
);
}
else
{
}
else
{
...
@@ -533,11 +546,13 @@ void SetTensorFromPyArray(framework::Tensor *self, const py::object &obj,
...
@@ -533,11 +546,13 @@ void SetTensorFromPyArray(framework::Tensor *self, const py::object &obj,
}
}
template
<
typename
P
>
template
<
typename
P
>
void
SetStringTensorFromPyArray
(
phi
::
StringTensor
*
self
,
const
py
::
array
&
array
,
void
SetStringTensorFromPyArray
(
phi
::
StringTensor
*
self
,
const
py
::
array
&
array
,
const
P
&
place
)
{
const
P
&
place
)
{
bool
is_string_pyarray
=
bool
is_string_pyarray
=
array
.
dtype
().
kind
()
==
'S'
||
array
.
dtype
().
kind
()
==
'U'
;
array
.
dtype
().
kind
()
==
'S'
||
array
.
dtype
().
kind
()
==
'U'
;
PADDLE_ENFORCE_EQ
(
is_string_pyarray
,
true
,
PADDLE_ENFORCE_EQ
(
is_string_pyarray
,
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"Expect the dtype of numpy array is string or "
"Expect the dtype of numpy array is string or "
"unicode, but recevie dtype %s"
,
"unicode, but recevie dtype %s"
,
...
@@ -575,7 +590,8 @@ void SetStringTensorFromPyArray(phi::StringTensor *self, const py::array &array,
...
@@ -575,7 +590,8 @@ void SetStringTensorFromPyArray(phi::StringTensor *self, const py::array &array,
pstring
pstr
(
utf8_len
-
1
,
0
);
pstring
pstr
(
utf8_len
-
1
,
0
);
phi
::
strings
::
GetUTF8Str
(
phi
::
strings
::
GetUTF8Str
(
reinterpret_cast
<
const
uint32_t
*>
(
array
.
data
())
+
unicode_len
*
i
,
reinterpret_cast
<
const
uint32_t
*>
(
array
.
data
())
+
unicode_len
*
i
,
pstr
.
mdata
(),
unicode_len
);
pstr
.
mdata
(),
unicode_len
);
dst
[
i
]
=
pstr
;
dst
[
i
]
=
pstr
;
}
}
}
}
...
@@ -588,7 +604,8 @@ void SetStringTensorFromPyArray(phi::StringTensor *self, const py::array &array,
...
@@ -588,7 +604,8 @@ void SetStringTensorFromPyArray(phi::StringTensor *self, const py::array &array,
template
<
typename
T
>
template
<
typename
T
>
void
SetUVATensorFromPyArrayImpl
(
framework
::
LoDTensor
*
self_tensor
,
void
SetUVATensorFromPyArrayImpl
(
framework
::
LoDTensor
*
self_tensor
,
const
py
::
array_t
<
T
>
&
array
,
int
device_id
)
{
const
py
::
array_t
<
T
>
&
array
,
int
device_id
)
{
#if defined(PADDLE_WITH_CUDA)
#if defined(PADDLE_WITH_CUDA)
VLOG
(
4
)
<<
"Running in SetUVATensorFromPyArrayImpl."
;
VLOG
(
4
)
<<
"Running in SetUVATensorFromPyArrayImpl."
;
std
::
vector
<
int64_t
>
dims
;
std
::
vector
<
int64_t
>
dims
;
...
@@ -603,16 +620,19 @@ void SetUVATensorFromPyArrayImpl(framework::LoDTensor *self_tensor,
...
@@ -603,16 +620,19 @@ void SetUVATensorFromPyArrayImpl(framework::LoDTensor *self_tensor,
auto
data_type
=
framework
::
ToDataType
(
std
::
type_index
(
typeid
(
T
)));
auto
data_type
=
framework
::
ToDataType
(
std
::
type_index
(
typeid
(
T
)));
const
auto
&
need_allocate_size
=
numel
*
framework
::
SizeOfType
(
data_type
);
const
auto
&
need_allocate_size
=
numel
*
framework
::
SizeOfType
(
data_type
);
T
*
data_ptr
;
T
*
data_ptr
;
cudaHostAlloc
(
reinterpret_cast
<
void
**>
(
&
data_ptr
),
need_allocate_size
,
cudaHostAlloc
(
reinterpret_cast
<
void
**>
(
&
data_ptr
),
need_allocate_size
,
cudaHostAllocWriteCombined
|
cudaHostAllocMapped
);
cudaHostAllocWriteCombined
|
cudaHostAllocMapped
);
std
::
memcpy
(
data_ptr
,
array
.
data
(),
array
.
nbytes
());
std
::
memcpy
(
data_ptr
,
array
.
data
(),
array
.
nbytes
());
void
*
cuda_device_pointer
=
nullptr
;
void
*
cuda_device_pointer
=
nullptr
;
cudaHostGetDevicePointer
(
reinterpret_cast
<
void
**>
(
&
cuda_device_pointer
),
cudaHostGetDevicePointer
(
reinterpret_cast
<
void
**>
(
&
cuda_device_pointer
),
reinterpret_cast
<
void
*>
(
data_ptr
),
0
);
reinterpret_cast
<
void
*>
(
data_ptr
),
0
);
std
::
shared_ptr
<
memory
::
allocation
::
Allocation
>
holder
=
std
::
shared_ptr
<
memory
::
allocation
::
Allocation
>
holder
=
std
::
make_shared
<
memory
::
allocation
::
Allocation
>
(
std
::
make_shared
<
memory
::
allocation
::
Allocation
>
(
cuda_device_pointer
,
need_allocate_size
,
cuda_device_pointer
,
need_allocate_size
,
platform
::
CUDAPlace
(
device_id
));
platform
::
CUDAPlace
(
device_id
));
self_tensor
->
ResetHolderWithType
(
holder
,
self_tensor
->
ResetHolderWithType
(
holder
,
framework
::
TransToPhiDataType
(
data_type
));
framework
::
TransToPhiDataType
(
data_type
));
...
@@ -622,7 +642,8 @@ void SetUVATensorFromPyArrayImpl(framework::LoDTensor *self_tensor,
...
@@ -622,7 +642,8 @@ void SetUVATensorFromPyArrayImpl(framework::LoDTensor *self_tensor,
template
<
typename
T
>
template
<
typename
T
>
void
SetUVATensorFromPyArray
(
void
SetUVATensorFromPyArray
(
const
std
::
shared_ptr
<
paddle
::
imperative
::
VarBase
>
&
self
,
const
std
::
shared_ptr
<
paddle
::
imperative
::
VarBase
>
&
self
,
const
py
::
array_t
<
T
>
&
array
,
int
device_id
)
{
const
py
::
array_t
<
T
>
&
array
,
int
device_id
)
{
#if defined(PADDLE_WITH_CUDA)
#if defined(PADDLE_WITH_CUDA)
VLOG
(
4
)
<<
"Running in SetUVATensorFromPyArray for VarBase."
;
VLOG
(
4
)
<<
"Running in SetUVATensorFromPyArray for VarBase."
;
auto
*
self_tensor
=
self
->
MutableVar
()
->
GetMutable
<
framework
::
LoDTensor
>
();
auto
*
self_tensor
=
self
->
MutableVar
()
->
GetMutable
<
framework
::
LoDTensor
>
();
...
@@ -633,7 +654,8 @@ void SetUVATensorFromPyArray(
...
@@ -633,7 +654,8 @@ void SetUVATensorFromPyArray(
template
<
typename
T
>
template
<
typename
T
>
void
SetUVATensorFromPyArray
(
void
SetUVATensorFromPyArray
(
const
std
::
shared_ptr
<
paddle
::
experimental
::
Tensor
>
&
self
,
const
std
::
shared_ptr
<
paddle
::
experimental
::
Tensor
>
&
self
,
const
py
::
array_t
<
T
>
&
array
,
int
device_id
)
{
const
py
::
array_t
<
T
>
&
array
,
int
device_id
)
{
#if defined(PADDLE_WITH_CUDA)
#if defined(PADDLE_WITH_CUDA)
VLOG
(
4
)
<<
"Running in SetUVATensorFromPyArray for Phi::Tensor."
;
VLOG
(
4
)
<<
"Running in SetUVATensorFromPyArray for Phi::Tensor."
;
phi
::
DenseTensorMeta
meta
=
phi
::
DenseTensorMeta
meta
=
...
@@ -652,7 +674,8 @@ void SetUVATensorFromPyArray(
...
@@ -652,7 +674,8 @@ void SetUVATensorFromPyArray(
}
}
template
<
typename
T
,
size_t
D
>
template
<
typename
T
,
size_t
D
>
void
_sliceCompute
(
const
framework
::
Tensor
*
in
,
framework
::
Tensor
*
out
,
void
_sliceCompute
(
const
framework
::
Tensor
*
in
,
framework
::
Tensor
*
out
,
const
platform
::
CPUDeviceContext
&
ctx
,
const
platform
::
CPUDeviceContext
&
ctx
,
const
std
::
vector
<
int
>
&
axes
,
const
std
::
vector
<
int
>
&
axes
,
const
std
::
vector
<
int
>
&
starts
)
{
const
std
::
vector
<
int
>
&
starts
)
{
...
@@ -688,15 +711,21 @@ void _sliceCompute(const framework::Tensor *in, framework::Tensor *out,
...
@@ -688,15 +711,21 @@ void _sliceCompute(const framework::Tensor *in, framework::Tensor *out,
template
<
typename
T
>
template
<
typename
T
>
void
_concatCompute
(
const
std
::
vector
<
paddle
::
framework
::
Tensor
>
&
ins
,
void
_concatCompute
(
const
std
::
vector
<
paddle
::
framework
::
Tensor
>
&
ins
,
paddle
::
framework
::
Tensor
*
out
,
paddle
::
framework
::
Tensor
*
out
,
const
platform
::
CPUDeviceContext
&
ctx
,
int64_t
axis
)
{
const
platform
::
CPUDeviceContext
&
ctx
,
int64_t
axis
)
{
if
(
axis
==
0
&&
ins
.
size
()
<
10
)
{
if
(
axis
==
0
&&
ins
.
size
()
<
10
)
{
size_t
output_offset
=
0
;
size_t
output_offset
=
0
;
for
(
auto
&
in
:
ins
)
{
for
(
auto
&
in
:
ins
)
{
auto
in_stride
=
phi
::
stride_numel
(
in
.
dims
());
auto
in_stride
=
phi
::
stride_numel
(
in
.
dims
());
auto
out_stride
=
phi
::
stride_numel
(
out
->
dims
());
auto
out_stride
=
phi
::
stride_numel
(
out
->
dims
());
paddle
::
operators
::
StridedNumelCopyWithAxis
<
T
>
(
paddle
::
operators
::
StridedNumelCopyWithAxis
<
T
>
(
ctx
,
axis
,
out
->
data
<
T
>
()
+
output_offset
,
out_stride
,
in
.
data
<
T
>
(),
ctx
,
in_stride
,
in_stride
[
axis
]);
axis
,
out
->
data
<
T
>
()
+
output_offset
,
out_stride
,
in
.
data
<
T
>
(),
in_stride
,
in_stride
[
axis
]);
output_offset
+=
in_stride
[
axis
];
output_offset
+=
in_stride
[
axis
];
}
}
}
else
{
}
else
{
...
@@ -706,9 +735,13 @@ void _concatCompute(const std::vector<paddle::framework::Tensor> &ins,
...
@@ -706,9 +735,13 @@ void _concatCompute(const std::vector<paddle::framework::Tensor> &ins,
}
}
}
}
inline
void
_getSliceinfo
(
const
framework
::
Tensor
&
self
,
py
::
object
obj
,
inline
void
_getSliceinfo
(
const
framework
::
Tensor
&
self
,
const
int64_t
dim
,
int64_t
*
pstart
,
int64_t
*
pstop
,
py
::
object
obj
,
int64_t
*
pstep
,
int64_t
*
pslicelength
)
{
const
int64_t
dim
,
int64_t
*
pstart
,
int64_t
*
pstop
,
int64_t
*
pstep
,
int64_t
*
pslicelength
)
{
auto
&
start
=
*
pstart
;
auto
&
start
=
*
pstart
;
auto
&
stop
=
*
pstop
;
auto
&
stop
=
*
pstop
;
auto
&
step
=
*
pstep
;
auto
&
step
=
*
pstep
;
...
@@ -718,7 +751,8 @@ inline void _getSliceinfo(const framework::Tensor &self, py::object obj,
...
@@ -718,7 +751,8 @@ inline void _getSliceinfo(const framework::Tensor &self, py::object obj,
0
<=
dim
&&
dim
<
srcDDim
.
size
(),
0
<=
dim
&&
dim
<
srcDDim
.
size
(),
platform
::
errors
::
OutOfRange
(
"The dim %d of slice is out of bounds, it "
platform
::
errors
::
OutOfRange
(
"The dim %d of slice is out of bounds, it "
"shound be in the range of [0, %d)."
,
"shound be in the range of [0, %d)."
,
dim
,
srcDDim
.
size
()));
dim
,
srcDDim
.
size
()));
if
(
py
::
isinstance
<
py
::
slice
>
(
obj
))
{
if
(
py
::
isinstance
<
py
::
slice
>
(
obj
))
{
size_t
lstart
,
lstop
,
lstep
,
lslicelength
;
size_t
lstart
,
lstop
,
lstep
,
lslicelength
;
...
@@ -739,7 +773,9 @@ inline void _getSliceinfo(const framework::Tensor &self, py::object obj,
...
@@ -739,7 +773,9 @@ inline void _getSliceinfo(const framework::Tensor &self, py::object obj,
std
::
abs
(
start
)
<
srcDDim
[
dim
],
std
::
abs
(
start
)
<
srcDDim
[
dim
],
platform
::
errors
::
OutOfRange
(
"The start %d of slice is out of bounds, "
platform
::
errors
::
OutOfRange
(
"The start %d of slice is out of bounds, "
"it shound be in the range of (%d, %d)."
,
"it shound be in the range of (%d, %d)."
,
start
,
-
srcDDim
[
dim
],
srcDDim
[
dim
]));
start
,
-
srcDDim
[
dim
],
srcDDim
[
dim
]));
start
=
(
start
>=
0
)
?
start
:
srcDDim
[
dim
]
-
start
;
start
=
(
start
>=
0
)
?
start
:
srcDDim
[
dim
]
-
start
;
stop
=
start
+
1
;
stop
=
start
+
1
;
step
=
1
;
step
=
1
;
...
@@ -779,9 +815,11 @@ inline framework::Tensor *_getTensor(const framework::Tensor &self,
...
@@ -779,9 +815,11 @@ inline framework::Tensor *_getTensor(const framework::Tensor &self,
}
}
template
<
typename
T
>
template
<
typename
T
>
void
_sliceDapper
(
const
framework
::
Tensor
*
in
,
framework
::
Tensor
*
out
,
void
_sliceDapper
(
const
framework
::
Tensor
*
in
,
framework
::
Tensor
*
out
,
const
platform
::
CPUDeviceContext
&
ctx
,
const
platform
::
CPUDeviceContext
&
ctx
,
const
std
::
vector
<
int
>
&
axes
,
const
std
::
vector
<
int
>
&
starts
,
const
std
::
vector
<
int
>
&
axes
,
const
std
::
vector
<
int
>
&
starts
,
int
size
)
{
int
size
)
{
switch
(
size
)
{
switch
(
size
)
{
case
1
:
case
1
:
...
@@ -821,7 +859,9 @@ void _sliceDapper(const framework::Tensor *in, framework::Tensor *out,
...
@@ -821,7 +859,9 @@ void _sliceDapper(const framework::Tensor *in, framework::Tensor *out,
template
<
typename
T
>
template
<
typename
T
>
inline
framework
::
Tensor
*
_sliceWrapper
(
const
framework
::
Tensor
&
self
,
inline
framework
::
Tensor
*
_sliceWrapper
(
const
framework
::
Tensor
&
self
,
const
platform
::
CPUDeviceContext
&
ctx
,
const
platform
::
CPUDeviceContext
&
ctx
,
py
::
object
obj
,
int
dim
,
int64_t
start
,
py
::
object
obj
,
int
dim
,
int64_t
start
,
int64_t
slicelength
)
{
int64_t
slicelength
)
{
framework
::
DDim
dstDDim
=
self
.
dims
();
framework
::
DDim
dstDDim
=
self
.
dims
();
dstDDim
[
dim
]
=
static_cast
<
int64_t
>
(
slicelength
);
dstDDim
[
dim
]
=
static_cast
<
int64_t
>
(
slicelength
);
...
@@ -834,7 +874,8 @@ inline framework::Tensor *_sliceWrapper(const framework::Tensor &self,
...
@@ -834,7 +874,8 @@ inline framework::Tensor *_sliceWrapper(const framework::Tensor &self,
template
<
typename
T
>
template
<
typename
T
>
inline
framework
::
Tensor
*
_sliceAndConcat
(
const
framework
::
Tensor
&
self
,
inline
framework
::
Tensor
*
_sliceAndConcat
(
const
framework
::
Tensor
&
self
,
py
::
object
obj
,
int
dim
)
{
py
::
object
obj
,
int
dim
)
{
platform
::
CPUDeviceContext
ctx
;
platform
::
CPUDeviceContext
ctx
;
int64_t
start
,
stop
,
step
,
slicelength
;
int64_t
start
,
stop
,
step
,
slicelength
;
_getSliceinfo
(
self
,
obj
,
dim
,
&
start
,
&
stop
,
&
step
,
&
slicelength
);
_getSliceinfo
(
self
,
obj
,
dim
,
&
start
,
&
stop
,
&
step
,
&
slicelength
);
...
@@ -856,7 +897,8 @@ inline framework::Tensor *_sliceAndConcat(const framework::Tensor &self,
...
@@ -856,7 +897,8 @@ inline framework::Tensor *_sliceAndConcat(const framework::Tensor &self,
}
}
inline
framework
::
Tensor
*
_sliceTensor
(
const
framework
::
Tensor
&
self
,
inline
framework
::
Tensor
*
_sliceTensor
(
const
framework
::
Tensor
&
self
,
py
::
object
obj
,
int
dim
)
{
py
::
object
obj
,
int
dim
)
{
auto
src_type
=
framework
::
TransToProtoVarType
(
self
.
dtype
());
auto
src_type
=
framework
::
TransToProtoVarType
(
self
.
dtype
());
switch
(
src_type
)
{
switch
(
src_type
)
{
case
framework
::
proto
::
VarType
::
FP16
:
case
framework
::
proto
::
VarType
::
FP16
:
...
@@ -959,43 +1001,53 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
...
@@ -959,43 +1001,53 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
!
is_custom_device_tensor
)
{
!
is_custom_device_tensor
)
{
if
(
!
need_deep_copy
)
{
if
(
!
need_deep_copy
)
{
auto
base
=
py
::
cast
(
std
::
move
(
tensor
));
auto
base
=
py
::
cast
(
std
::
move
(
tensor
));
return
py
::
array
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
,
return
py
::
array
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
const_cast
<
void
*>
(
tensor_buf_ptr
),
base
);
py_dims
,
py_strides
,
const_cast
<
void
*>
(
tensor_buf_ptr
),
base
);
}
else
{
}
else
{
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
PADDLE_ENFORCE_EQ
(
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
py_arr
.
writeable
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray is not writable, in which case memory leak "
"PyArray is not writable, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
PADDLE_ENFORCE_EQ
(
PADDLE_ENFORCE_EQ
(
py_arr
.
owndata
(),
true
,
py_arr
.
owndata
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray does not own data, in which case memory leak "
"PyArray does not own data, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
platform
::
CPUPlace
place
;
platform
::
CPUPlace
place
;
size_t
copy_bytes
=
sizeof_dtype
*
numel
;
size_t
copy_bytes
=
sizeof_dtype
*
numel
;
paddle
::
memory
::
Copy
(
place
,
py_arr
.
mutable_data
(),
place
,
tensor_buf_ptr
,
paddle
::
memory
::
Copy
(
copy_bytes
);
place
,
py_arr
.
mutable_data
(),
place
,
tensor_buf_ptr
,
copy_bytes
);
return
py_arr
;
return
py_arr
;
}
}
}
else
if
(
is_xpu_tensor
)
{
}
else
if
(
is_xpu_tensor
)
{
#ifdef PADDLE_WITH_XPU
#ifdef PADDLE_WITH_XPU
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray is not writable, in which case memory leak "
"PyArray is not writable, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
PADDLE_ENFORCE_EQ
(
PADDLE_ENFORCE_EQ
(
py_arr
.
owndata
(),
true
,
py_arr
.
owndata
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray does not own data, in which case memory leak "
"PyArray does not own data, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
size_t
copy_bytes
=
sizeof_dtype
*
numel
;
size_t
copy_bytes
=
sizeof_dtype
*
numel
;
auto
p
=
tensor
.
place
();
auto
p
=
tensor
.
place
();
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
py_arr
.
mutable_data
(),
p
,
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
tensor_buf_ptr
,
copy_bytes
);
py_arr
.
mutable_data
(),
p
,
tensor_buf_ptr
,
copy_bytes
);
return
py_arr
;
return
py_arr
;
#else
#else
PADDLE_THROW
(
platform
::
errors
::
PermissionDenied
(
PADDLE_THROW
(
platform
::
errors
::
PermissionDenied
(
...
@@ -1005,20 +1057,26 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
...
@@ -1005,20 +1057,26 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
}
else
if
(
is_gpu_tensor
)
{
}
else
if
(
is_gpu_tensor
)
{
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray is not writable, in which case memory leak "
"PyArray is not writable, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
PADDLE_ENFORCE_EQ
(
PADDLE_ENFORCE_EQ
(
py_arr
.
owndata
(),
true
,
py_arr
.
owndata
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray does not own data, in which case memory leak "
"PyArray does not own data, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
size_t
copy_bytes
=
sizeof_dtype
*
numel
;
size_t
copy_bytes
=
sizeof_dtype
*
numel
;
auto
p
=
tensor
.
place
();
auto
p
=
tensor
.
place
();
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
py_arr
.
mutable_data
(),
p
,
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
tensor_buf_ptr
,
copy_bytes
,
nullptr
);
py_arr
.
mutable_data
(),
p
,
tensor_buf_ptr
,
copy_bytes
,
nullptr
);
return
py_arr
;
return
py_arr
;
#else
#else
PADDLE_THROW
(
platform
::
errors
::
PermissionDenied
(
PADDLE_THROW
(
platform
::
errors
::
PermissionDenied
(
...
@@ -1028,12 +1086,14 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
...
@@ -1028,12 +1086,14 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
}
else
if
(
is_npu_tensor
)
{
}
else
if
(
is_npu_tensor
)
{
#ifdef PADDLE_WITH_ASCEND_CL
#ifdef PADDLE_WITH_ASCEND_CL
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray is not writable, in which case memory leak "
"PyArray is not writable, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
PADDLE_ENFORCE_EQ
(
PADDLE_ENFORCE_EQ
(
py_arr
.
owndata
(),
true
,
py_arr
.
owndata
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray does not own data, in which case memory leak "
"PyArray does not own data, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
...
@@ -1043,7 +1103,10 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
...
@@ -1043,7 +1103,10 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
platform
::
DeviceContextPool
&
pool
=
platform
::
DeviceContextPool
::
Instance
();
platform
::
DeviceContextPool
&
pool
=
platform
::
DeviceContextPool
::
Instance
();
auto
&
ctx
=
*
pool
.
Get
(
tensor
.
place
());
auto
&
ctx
=
*
pool
.
Get
(
tensor
.
place
());
paddle
::
memory
::
Copy
(
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
py_arr
.
mutable_data
(),
p
,
tensor_buf_ptr
,
platform
::
CPUPlace
(),
py_arr
.
mutable_data
(),
p
,
tensor_buf_ptr
,
copy_bytes
,
copy_bytes
,
reinterpret_cast
<
const
platform
::
NPUDeviceContext
&>
(
ctx
).
stream
());
reinterpret_cast
<
const
platform
::
NPUDeviceContext
&>
(
ctx
).
stream
());
ctx
.
Wait
();
ctx
.
Wait
();
...
@@ -1056,12 +1119,14 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
...
@@ -1056,12 +1119,14 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
}
else
if
(
is_mlu_tensor
)
{
}
else
if
(
is_mlu_tensor
)
{
#ifdef PADDLE_WITH_MLU
#ifdef PADDLE_WITH_MLU
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray is not writable, in which case memory leak "
"PyArray is not writable, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
PADDLE_ENFORCE_EQ
(
PADDLE_ENFORCE_EQ
(
py_arr
.
owndata
(),
true
,
py_arr
.
owndata
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray does not own data, in which case memory leak "
"PyArray does not own data, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
...
@@ -1071,7 +1136,10 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
...
@@ -1071,7 +1136,10 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
platform
::
DeviceContextPool
&
pool
=
platform
::
DeviceContextPool
::
Instance
();
platform
::
DeviceContextPool
&
pool
=
platform
::
DeviceContextPool
::
Instance
();
auto
&
ctx
=
*
pool
.
Get
(
tensor
.
place
());
auto
&
ctx
=
*
pool
.
Get
(
tensor
.
place
());
paddle
::
memory
::
Copy
(
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
py_arr
.
mutable_data
(),
p
,
tensor_buf_ptr
,
platform
::
CPUPlace
(),
py_arr
.
mutable_data
(),
p
,
tensor_buf_ptr
,
copy_bytes
,
copy_bytes
,
reinterpret_cast
<
const
platform
::
MLUDeviceContext
&>
(
ctx
).
stream
());
reinterpret_cast
<
const
platform
::
MLUDeviceContext
&>
(
ctx
).
stream
());
ctx
.
Wait
();
ctx
.
Wait
();
...
@@ -1084,12 +1152,14 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
...
@@ -1084,12 +1152,14 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
}
else
if
(
is_custom_device_tensor
)
{
}
else
if
(
is_custom_device_tensor
)
{
#ifdef PADDLE_WITH_CUSTOM_DEVICE
#ifdef PADDLE_WITH_CUSTOM_DEVICE
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
py
::
array
py_arr
(
py
::
dtype
(
py_dtype_str
.
c_str
()),
py_dims
,
py_strides
);
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
PADDLE_ENFORCE_EQ
(
py_arr
.
writeable
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray is not writable, in which case memory leak "
"PyArray is not writable, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
PADDLE_ENFORCE_EQ
(
PADDLE_ENFORCE_EQ
(
py_arr
.
owndata
(),
true
,
py_arr
.
owndata
(),
true
,
platform
::
errors
::
InvalidArgument
(
platform
::
errors
::
InvalidArgument
(
"PyArray does not own data, in which case memory leak "
"PyArray does not own data, in which case memory leak "
"or double free would occur"
));
"or double free would occur"
));
...
@@ -1098,8 +1168,11 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
...
@@ -1098,8 +1168,11 @@ inline py::array TensorToPyArray(const framework::Tensor &tensor,
platform
::
DeviceContextPool
&
pool
=
platform
::
DeviceContextPool
::
Instance
();
platform
::
DeviceContextPool
&
pool
=
platform
::
DeviceContextPool
::
Instance
();
auto
&
ctx
=
*
pool
.
Get
(
tensor
.
place
());
auto
&
ctx
=
*
pool
.
Get
(
tensor
.
place
());
paddle
::
memory
::
Copy
(
paddle
::
memory
::
Copy
(
platform
::
CPUPlace
(),
py_arr
.
mutable_data
(),
tensor
.
place
(),
platform
::
CPUPlace
(),
tensor_buf_ptr
,
copy_bytes
,
py_arr
.
mutable_data
(),
tensor
.
place
(),
tensor_buf_ptr
,
copy_bytes
,
reinterpret_cast
<
const
platform
::
CustomDeviceContext
&>
(
ctx
).
stream
());
reinterpret_cast
<
const
platform
::
CustomDeviceContext
&>
(
ctx
).
stream
());
ctx
.
Wait
();
ctx
.
Wait
();
return
py_arr
;
return
py_arr
;
...
...
paddle/phi/common/data_type.h
浏览文件 @
66a28e13
...
@@ -18,7 +18,12 @@ limitations under the License. */
...
@@ -18,7 +18,12 @@ limitations under the License. */
#include "paddle/phi/common/bfloat16.h"
#include "paddle/phi/common/bfloat16.h"
#include "paddle/phi/common/complex.h"
#include "paddle/phi/common/complex.h"
#include "paddle/phi/common/float16.h"
#include "paddle/phi/common/float16.h"
#include "paddle/phi/common/pstring.h"
namespace
phi
{
namespace
dtype
{
class
pstring
;
}
// namespace dtype
}
// namespace phi
namespace
paddle
{
namespace
paddle
{
namespace
experimental
{
namespace
experimental
{
...
...
paddle/phi/core/string_tensor.cc
浏览文件 @
66a28e13
...
@@ -15,6 +15,7 @@ limitations under the License. */
...
@@ -15,6 +15,7 @@ limitations under the License. */
#include "paddle/phi/core/string_tensor.h"
#include "paddle/phi/core/string_tensor.h"
#include "paddle/fluid/memory/malloc.h"
#include "paddle/fluid/memory/malloc.h"
#include "paddle/phi/common/pstring.h"
namespace
phi
{
namespace
phi
{
...
...
paddle/phi/core/string_tensor.h
浏览文件 @
66a28e13
...
@@ -14,13 +14,16 @@ limitations under the License. */
...
@@ -14,13 +14,16 @@ limitations under the License. */
#pragma once
#pragma once
#include "paddle/phi/common/pstring.h"
#include "paddle/phi/core/allocator.h"
#include "paddle/phi/core/allocator.h"
#include "paddle/phi/core/tensor_base.h"
#include "paddle/phi/core/tensor_base.h"
#include "paddle/phi/core/tensor_meta.h"
#include "paddle/phi/core/tensor_meta.h"
namespace
phi
{
namespace
phi
{
namespace
dtype
{
class
pstring
;
}
// namespace dtype
/// \brief In Paddle 2.3, we add a new type of Tensor, StringTensor,
/// \brief In Paddle 2.3, we add a new type of Tensor, StringTensor,
/// which is designed for string data management.
/// which is designed for string data management.
/// During the entire life cycle of a StringTensor, its device type and key
/// During the entire life cycle of a StringTensor, its device type and key
...
...
paddle/phi/kernels/strings/cpu/strings_copy_kernel.cc
浏览文件 @
66a28e13
...
@@ -15,6 +15,7 @@ limitations under the License. */
...
@@ -15,6 +15,7 @@ limitations under the License. */
#include "paddle/phi/kernels/strings/strings_copy_kernel.h"
#include "paddle/phi/kernels/strings/strings_copy_kernel.h"
#include "glog/logging.h"
#include "glog/logging.h"
#include "paddle/phi/common/pstring.h"
#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/core/kernel_registry.h"
namespace
phi
{
namespace
phi
{
...
...
paddle/phi/tests/api/test_strings_lower_upper_api.cc
浏览文件 @
66a28e13
...
@@ -19,6 +19,7 @@ limitations under the License. */
...
@@ -19,6 +19,7 @@ limitations under the License. */
#include "paddle/phi/api/include/strings_api.h"
#include "paddle/phi/api/include/strings_api.h"
#include "paddle/phi/api/lib/utils/allocator.h"
#include "paddle/phi/api/lib/utils/allocator.h"
#include "paddle/phi/backends/all_context.h"
#include "paddle/phi/backends/all_context.h"
#include "paddle/phi/common/pstring.h"
#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/core/string_tensor.h"
#include "paddle/phi/core/string_tensor.h"
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录