fix_gru_py
yi.wu 7 years ago
parent 7a7d27b33e
commit 6ef60de6f1

@ -1,7 +1,7 @@
#!/bin/bash
set -e
readonly VERSION="3.8"
readonly VERSION="7.0"
version=$(clang-format -version)

@ -162,7 +162,7 @@ void SerializeToByteBuffer(const std::string& name, framework::Variable* var,
if (var->IsType<ncclUniqueId>()) {
e.WriteVarlengthBeginning(VarMsg::kSerializedFieldNumber,
NCCL_UNIQUE_ID_BYTES);
ncclUniqueId& uid = var->Get<ncclUniqueId>();
const ncclUniqueId& uid = var->Get<ncclUniqueId>();
e.WriteRawBytes(std::string(uid.internal, NCCL_UNIQUE_ID_BYTES));
// for serialize NCCL_ID

@ -23,6 +23,7 @@ limitations under the License. */
#include "paddle/fluid/framework/threadpool.h"
#include "paddle/fluid/operators/detail/grpc_client.h"
#include "paddle/fluid/operators/detail/grpc_server.h"
#include "paddle/fluid/platform/nccl_helper.h"
namespace paddle {
namespace operators {

Loading…
Cancel
Save