shpark commented on a change in pull request #81: [NEMO-151] Add OutputWriters
for additional tagged outputs
URL: https://github.com/apache/incubator-nemo/pull/81#discussion_r203945218
##########
File path:
compiler/optimizer/src/main/java/edu/snu/nemo/compiler/optimizer/pass/compiletime/reshaping/SailfishRelayReshapingPass.java
##########
@@ -60,6 +62,21 @@ public SailfishRelayReshapingPass() {
// before the vertex receiving shuffled data.
final OperatorVertex iFileMergerVertex = new OperatorVertex(new
RelayTransform());
iFileMergerVertex.getExecutionProperties().put(SkipSerDesProperty.of());
+
+ // Update additional tagged output property of the source vertex
if necessary.
+ if
(edge.getSrc().getPropertyValue(AdditionalTagOutputProperty.class).isPresent())
{
+ final HashMap<String, String> additionalTagOutputPropertyValue =
+
edge.getSrc().getPropertyValue(AdditionalTagOutputProperty.class).get();
+ if (additionalTagOutputPropertyValue.containsValue(v.getId())) {
+ additionalTagOutputPropertyValue.forEach((tag, vertexId) -> {
+ if (vertexId == v.getId()) {
Review comment:
Thanks for your comments. I will fix this.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
