Change to PYBIND11_MODULE because the deprecation of PYBIND11_PLUGIN

test=develop
local_add_cudnn_lstm
minqiyang 7 years ago
parent 81994e84e0
commit b67229187e

@ -86,7 +86,7 @@ bool IsCompiledWithDIST() {
#endif
}
PYBIND11_PLUGIN(core) {
PYBIND11_MODULE(core) {
// Not used, just make sure cpu_info.cc is linked.
paddle::platform::CpuTotalPhysicalMemory();

Loading…
Cancel
Save