Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!31172 [FL][master] signds clean code 

Merge pull request !31172 from emmmmtang/master
This commit is contained in:
i-robot 2022-03-14 04:53:32 +00:00 committed by Gitee
parent 9db67a72cc 64bbc4813a
commit f8d5ffe48a
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 4 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/ccsrc/fl/server/kernel/round/update_model_kernel.cc
View File

@ -356,15 +356,7 @@ std::map<std::string, UploadData> UpdateModelKernel::ParseSignDSFeatureMap(
std::map<std::string, UploadData> feature_map; std::map<std::string, UploadData> feature_map;
auto sign = update_model_req->sign(); auto sign = update_model_req->sign();
if (sign == 0) { if (sign == 0) {
for (uint32_t i = 0; i < fbs_feature_map->size(); i++) { feature_map = ParseFeatureMap(update_model_req);
std::string weight_full_name = fbs_feature_map->Get(i)->weight_fullname()->str();
float *weight_data = const_cast<float *>(fbs_feature_map->Get(i)->data()->data());
size_t weight_size = fbs_feature_map->Get(i)->data()->size() * sizeof(float);
UploadData upload_data;
upload_data[kNewWeight].addr = weight_data;
upload_data[kNewWeight].size = weight_size;
feature_map[weight_full_name] = upload_data;
}
return feature_map; return feature_map;
} }

6
mindspore/lite/java/java/fl_client/src/main/java/com/mindspore/flclient/UpdateModel.java
View File

@ -320,7 +320,7 @@ public class UpdateModel {
this.fmOffset = RequestUpdateModel.createFeatureMapVector(builder, fmOffsetsDP); this.fmOffset = RequestUpdateModel.createFeatureMapVector(builder, fmOffsetsDP);
LOGGER.info(Common.addTag("[Encrypt] DP mask model ok!")); LOGGER.info(Common.addTag("[Encrypt] DP mask model ok!"));
endTime = System.currentTimeMillis(); endTime = System.currentTimeMillis();
LOGGER.info(Common.addTag("dp time is：" + (endTime - startTime) + "ms")); LOGGER.info(Common.addTag("[Encrypt] dp time is: " + (endTime - startTime) + "ms"));
return this; return this;
case SIGNDS: case SIGNDS:
startTime = System.currentTimeMillis(); startTime = System.currentTimeMillis();
@ -352,7 +352,7 @@ public class UpdateModel {
this.fmOffset = RequestUpdateModel.createFeatureMapVector(builder, fmOffsetsSignds); this.fmOffset = RequestUpdateModel.createFeatureMapVector(builder, fmOffsetsSignds);
LOGGER.info(Common.addTag("[Encrypt] SignDS mask model ok!")); LOGGER.info(Common.addTag("[Encrypt] SignDS mask model ok!"));
endTime = System.currentTimeMillis(); endTime = System.currentTimeMillis();
LOGGER.info(Common.addTag("signds time is：" + (endTime - startTime) + "ms")); LOGGER.info(Common.addTag("[Encrypt] signds time is: " + (endTime - startTime) + "ms"));
return this; return this;
case NOT_ENCRYPT: case NOT_ENCRYPT:
default: default:
@ -374,7 +374,7 @@ public class UpdateModel {
} }
this.fmOffset = RequestUpdateModel.createFeatureMapVector(builder, fmOffsets); this.fmOffset = RequestUpdateModel.createFeatureMapVector(builder, fmOffsets);
endTime = System.currentTimeMillis(); endTime = System.currentTimeMillis();
LOGGER.info(Common.addTag("not encrypt time is：" + (endTime - startTime) + "ms")); LOGGER.info(Common.addTag("[Encrypt] not encrypt time is: " + (endTime - startTime) + "ms"));
return this; return this;
} }
} }