m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix code style of fleet_wrapper (#21639)

Browse Source 
* fix code style of fleet_wrapper
* test=develop
paddle_tiny_install
xujiaqi01 5 years ago committed by GitHub
parent 95b95a284b
commit c05706fe73
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File

6
paddle/fluid/framework/fleet/fleet_wrapper.cc
Unescape View File

@ -161,7 +161,7 @@ void FleetWrapper::PullSparseVarsSync(
LoDTensor* tensor = var->GetMutable<LoDTensor>(); LoDTensor* tensor = var->GetMutable<LoDTensor>();
CHECK(tensor != nullptr) << "tensor of var " << name << " is null"; CHECK(tensor != nullptr) << "tensor of var " << name << " is null";
int64_t* ids = tensor->data<int64_t>(); int64_t* ids = tensor->data<int64_t>();
int len = tensor->numel(); size_t len = tensor->numel();
// skip slots which do not have embedding // skip slots which do not have embedding
const std::string& emb_name = var_emb_names[var_index]; const std::string& emb_name = var_emb_names[var_index];
@ -350,7 +350,7 @@ void FleetWrapper::PushSparseVarsWithLabelAsync(
LOG(ERROR) << "tensor of var[" << sparse_key_names[i] << "] is null"; LOG(ERROR) << "tensor of var[" << sparse_key_names[i] << "] is null";
exit(-1); exit(-1);
} }
int len = tensor->numel(); size_t len = tensor->numel();
int64_t* ids = tensor->data<int64_t>(); int64_t* ids = tensor->data<int64_t>();
int slot = 0; int slot = 0;
if (dump_slot) { if (dump_slot) {
@ -413,7 +413,7 @@ void FleetWrapper::PushSparseVarsWithLabelAsync(
LOG(ERROR) << "tensor of var[" << sparse_key_names[i] << "] is null"; LOG(ERROR) << "tensor of var[" << sparse_key_names[i] << "] is null";
exit(-1); exit(-1);
} }
int len = tensor->numel(); size_t len = tensor->numel();
int64_t* ids = tensor->data<int64_t>(); int64_t* ids = tensor->data<int64_t>();
for (auto id_idx = 0u; id_idx < len; ++id_idx) { for (auto id_idx = 0u; id_idx < len; ++id_idx) {
if (ids[id_idx] == 0) { if (ids[id_idx] == 0) {

Write Preview
Loading…
Cancel
Save