Compare commits
79 Commits
test_bench
...
develop
| Author | SHA1 | Date |
|---|---|---|
wuhuanzhou
|
587d99ae44 | 5 years ago |
tianshuo78520a
|
b09c1ce09a | 5 years ago |
Thunderbrook
|
393b3bd6b7 | 5 years ago |
|
wuhuanzhou
|
3a95a0bc26 | 5 years ago |
taixiurong
|
52b05baca3 | 5 years ago |
Wenyu
|
5394194e3a | 5 years ago |
furnace
|
ef8323d49e | 5 years ago |
Jiawei Wang
|
95f808c878 | 5 years ago |
liym27
|
57d4288ad4 | 5 years ago |
石晓伟
|
0fa6c8a35c | 5 years ago |
Pei Yang
|
98e803e04f | 5 years ago |
|
wuhuanzhou
|
a37a7f67e1 | 5 years ago |
jakpiase
|
6dca7a1de7 | 5 years ago |
|
Pei Yang
|
14b7e3cf06 | 5 years ago |
Zhou Wei
|
245252b86e | 5 years ago |
Zhen Wang
|
e1f931610e | 5 years ago |
Yiqun Liu
|
e50bc2c2a6 | 5 years ago |
|
Zhou Wei
|
04a49b097e | 5 years ago |
wangguanzhong
|
fe2848686b | 5 years ago |
chajchaj
|
73a6fa3ed0 | 5 years ago |
Shang Zhizhou
|
8084b7594b | 5 years ago |
zlsh80826
|
64ee255ffd | 5 years ago |
Aurelius84
|
c4b60efabd | 5 years ago |
|
wuhuanzhou
|
17030ff28b | 5 years ago |
niuliling123
|
a71d72d921 | 5 years ago |
|
tianshuo78520a
|
8829a309fe | 5 years ago |
wanghuancoder
|
b48841ba2e | 5 years ago |
|
liym27
|
525c32e33c | 5 years ago |
ronnywang
|
123949eb48 | 5 years ago |
|
Shang Zhizhou
|
61805d8f0a | 5 years ago |
Jiabin Yang
|
51eb29de18 | 5 years ago |
|
zlsh80826
|
e3a38d790a | 5 years ago |
|
zlsh80826
|
bfb5cf5567 | 5 years ago |
cc
|
b47478efc2 | 5 years ago |
lilong12
|
c3974d0e2a | 5 years ago |
|
zlsh80826
|
01aa252624 | 5 years ago |
Wilber
|
70b67f1029 | 5 years ago |
|
tianshuo78520a
|
e804f08559 | 5 years ago |
Chengmo
|
f58cb01864 | 5 years ago |
Kaipeng Deng
|
bf09dcb346 | 5 years ago |
Chen Weihang
|
27f2d8df8e | 5 years ago |
|
Zhou Wei
|
511e204e62 | 5 years ago |
|
niuliling123
|
6472d62093 | 5 years ago |
winter-wang
|
e7f28d6c0d | 5 years ago |
parap1uie-s
|
5d89ec36dc | 5 years ago |
Huihuang Zheng
|
649868ffb2 | 5 years ago |
Wojciech Uss
|
e5f7a834d4 | 5 years ago |
|
Aurelius84
|
f2cfc0f46d | 5 years ago |
|
cc
|
84a551380e | 5 years ago |
|
Chen Weihang
|
68497e7b39 | 5 years ago |
|
ronnywang
|
270699e647 | 5 years ago |
|
Zhou Wei
|
1eb927f935 | 5 years ago |
|
Chen Weihang
|
3f66e7deab | 5 years ago |
Feiyu Chan
|
4046f1303a | 5 years ago |
|
Zhou Wei
|
a70de87d76 | 5 years ago |
|
Wilber
|
f4d9212de2 | 5 years ago |
|
niuliling123
|
372ac08a17 | 5 years ago |
|
Wojciech Uss
|
814b38e30f | 5 years ago |
|
tianshuo78520a
|
513641e153 | 5 years ago |
|
Shang Zhizhou
|
9d04ef7369 | 5 years ago |
Qi Li
|
46dd1d4aad | 5 years ago |
gongweibao
|
f72d197ec5 | 5 years ago |
Tao Luo
|
032de0bfd0 | 5 years ago |
|
zlsh80826
|
bfced39eb6 | 5 years ago |
arlesniak
|
7ccf6b6030 | 5 years ago |
|
lilong12
|
a501a7b0ca | 5 years ago |
guofei
|
ed7956a816 | 5 years ago |
|
ronnywang
|
8c19d7aa2f | 5 years ago |
Ouyang Chao
|
a45c8ca69d | 5 years ago |
Jacek Czaja
|
25fc2a1fdb | 5 years ago |
|
Chen Weihang
|
878e117b6d | 5 years ago |
|
ronnywang
|
c9e1d9dc31 | 5 years ago |
|
zlsh80826
|
1c67cf0c98 | 5 years ago |
|
Chen Weihang
|
e429deb0c4 | 5 years ago |
Adam Osewski
|
a4a2b77def | 5 years ago |
|
zlsh80826
|
c86e771e94 | 5 years ago |
|
zlsh80826
|
50cafa0b0c | 5 years ago |
|
cc
|
1d197f6c97 | 5 years ago |
|
ronnywang
|
420527f0d9 | 5 years ago |
@ -1,53 +0,0 @@
|
||||
/* Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License. */
|
||||
|
||||
#include "paddle/fluid/framework/c/c_api.h"
|
||||
|
||||
#include "paddle/fluid/framework/op_info.h"
|
||||
#include "paddle/fluid/platform/device_context.h"
|
||||
#include "paddle/fluid/platform/enforce.h"
|
||||
|
||||
extern "C" {
|
||||
|
||||
paddle::framework::OpInfoMap &PD_GetOpInfoMap() {
|
||||
return paddle::framework::OpInfoMap::Instance();
|
||||
}
|
||||
|
||||
void PD_InitDevicesPool(paddle::platform::DeviceContextPool *pool) {
|
||||
paddle::platform::DeviceContextPool::SetPool(pool);
|
||||
}
|
||||
|
||||
std::vector<std::string> PD_GetGradOpDescStrs(
|
||||
const paddle::framework::OpDesc &op_desc,
|
||||
const std::unordered_set<std::string> &no_grad_set,
|
||||
std::unordered_map<std::string, std::string> *grad_to_var,
|
||||
const std::vector<paddle::framework::BlockDesc *> &grad_block) {
|
||||
auto &op_info = PD_GetOpInfoMap().Get(op_desc.Type());
|
||||
std::vector<std::string> ret;
|
||||
if (op_info.grad_op_maker_) {
|
||||
auto grad_op_descs =
|
||||
op_info.grad_op_maker_(op_desc, no_grad_set, grad_to_var, grad_block);
|
||||
size_t op_num = grad_op_descs.size();
|
||||
ret.resize(op_num);
|
||||
for (size_t i = 0; i < op_num; ++i) {
|
||||
PADDLE_ENFORCE_EQ(
|
||||
grad_op_descs[i]->Proto()->SerializePartialToString(&ret[i]), true,
|
||||
paddle::platform::errors::Unavailable(
|
||||
"Cannot serialize operator desc message."));
|
||||
}
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
} // end extern "C"
|
||||
@ -1,55 +0,0 @@
|
||||
/* copyright (c) 2019 paddlepaddle authors. all rights reserved.
|
||||
|
||||
Licensed under the Apache License, Version 2.0 (the "License");
|
||||
you may not use this file except in compliance with the License.
|
||||
You may obtain a copy of the License at
|
||||
|
||||
http://www.apache.org/licenses/LICENSE-2.0
|
||||
|
||||
Unless required by applicable law or agreed to in writing, software
|
||||
distributed under the License is distributed on an "AS IS" BASIS,
|
||||
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License. */
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <string>
|
||||
#include <unordered_map>
|
||||
#include <unordered_set>
|
||||
#include <vector>
|
||||
|
||||
#include "paddle/fluid/framework/block_desc.h"
|
||||
#include "paddle/fluid/framework/op_desc.h"
|
||||
#include "paddle/fluid/framework/op_info.h"
|
||||
#include "paddle/fluid/platform/device_context.h"
|
||||
|
||||
namespace paddle {
|
||||
namespace framework {
|
||||
class OpInfoMap;
|
||||
} // namespace framework
|
||||
namespace platform {
|
||||
class DeviceContextPool;
|
||||
} // namespace platform
|
||||
} // namespace paddle
|
||||
|
||||
#ifdef __cplusplus
|
||||
extern "C" {
|
||||
#endif
|
||||
|
||||
// C-API to get global OpInfo map.
|
||||
paddle::framework::OpInfoMap &PD_GetOpInfoMap();
|
||||
|
||||
// C-API to init global DeviceContextPool from outside.
|
||||
void PD_InitDevicesPool(paddle::platform::DeviceContextPool *pool);
|
||||
|
||||
// C-API to serialize the grad op protocol message to a binary string.
|
||||
std::vector<std::string> PD_GetGradOpDescStrs(
|
||||
const paddle::framework::OpDesc &op_desc,
|
||||
const std::unordered_set<std::string> &no_grad_set,
|
||||
std::unordered_map<std::string, std::string> *grad_to_var,
|
||||
const std::vector<paddle::framework::BlockDesc *> &grad_block);
|
||||
|
||||
#ifdef __cplusplus
|
||||
}
|
||||
#endif
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue