Add sparse convertion api and sparse creation api #40780
Conversation
|
你的PR提交成功，感谢你对开源项目的贡献! |
zkh2016
changed the title
paddle/phi/api/lib/CMakeLists.txt
Outdated
| cc_library(sparse_api_custom_impl SRCS sparse_api_custom_impl.cc DEPS phi_tensor_raw phi kernel_dispatch api_gen_utils phi_data_transform) | ||
|
|
||
| cc_library(phi_function_api SRCS ${api_source_file} DEPS phi_tensor_raw phi kernel_dispatch api_gen_utils phi_data_transform api_custom_impl) | ||
| cc_library(phi_function_api SRCS ${api_source_file} DEPS phi_tensor_raw phi kernel_dispatch api_gen_utils phi_data_transform api_custom_impl sparse_api_custom_impl) |
There was a problem hiding this comment.
这里的依赖应该不需要加，普通的前向API不会依赖自定义的sparse_api
There was a problem hiding this comment.
已去掉，改成在编译tensor_methord.cc地方添加依赖。
| /* Part 11: Methods of converting SparseTensor and DenseTensor to each other | ||
| */ | ||
| /** | ||
| * @brief Convert DenseTensor or SparseCsrTensor to SparseCooTensor | ||
| * | ||
| * @param sparse_dim, The number of sparse dimensions | ||
| * @return Tensor | ||
| */ | ||
| Tensor to_sparse_coo(const int64_t sparse_dim); | ||
|
|
||
| /** | ||
| * @brief Convert DenseTensor or SparseCooTensor to SparseCsrTensor | ||
| * | ||
| * @return Tensor | ||
| */ | ||
| Tensor to_sparse_csr(); | ||
|
|
||
| /** | ||
| * @brief Convert SparseCooTensor or SparseCsrTensor to DenseTensor | ||
| * | ||
| * @return Tensor | ||
| */ | ||
| Tensor to_dense(); | ||
|
|
| return ToPyObject(coo_tensor); | ||
| EAGER_CATCH_AND_THROW_RETURN_NULL | ||
| } | ||
|
|
There was a problem hiding this comment.
I cannot find this method "tensor_method_to_sparse_coo" in
paddle\fluid\pybind\eager_method.cc on the "develop" branch.
There was a problem hiding this comment.
'to_sparse_coo' has been move to varbase_patch_methods.py
| return ToPyObject(dense_tensor); | ||
| EAGER_CATCH_AND_THROW_RETURN_NULL | ||
| } | ||
|
|
There was a problem hiding this comment.
Now, I cannot find "tensor_method_to_dense" in paddle\fluid\pybind\eager_method.cc on
the "develop" branch
There was a problem hiding this comment.
‘to_dense' has been move to varbase_patch_methods.py
4 participants
PR types
Others
PR changes
Others
Describe