Skip to content

Conversation

@Yancey0623
Copy link

Support compilation result Serialize and Deserialize.

Copy link

@anw90 anw90 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Please add some ut for this pr to check if the caching works

for (auto device : computation->devices()) {
result_pb.add_devices(device);
}
return hlo_proto.SerializeAsString() + ":::" + result_pb.SerializeAsString();
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Maybe we could use absl::strcat to concat the strings.

res.inputs = inputs;
res.outputs = outputs;

DISCCompileResult result;
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Remove this redundant code.

int DISCComputationClient::GetProcessIndex() const { return local_rank_; }

int DISCComputationClient::GetNumProcesses() const { return world_size_; }
std::string DISCComputationClient::SerializeComputation(
Copy link

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Add some blank lines between functions.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants