2626import org .apache .sysds .hops .Hop ;
2727import org .apache .sysds .hops .ReorgOp ;
2828import org .apache .sysds .parser .StatementBlock ;
29+ import org .apache .sysds .utils .ParameterizedLogger ;
2930
3031import java .util .ArrayList ;
3132import java .util .HashMap ;
5152 * {@code TeeOp}, and safely rewire the graph.
5253 */
5354public class RewriteInjectOOCTee extends StatementBlockRewriteRule {
55+ private static final ParameterizedLogger LOG = ParameterizedLogger .getLogger (RewriteInjectOOCTee .class );
5456
5557 public static boolean APPLY_ONLY_XtX_PATTERN = false ;
5658
@@ -138,10 +140,7 @@ private void applyTopDownTeeRewrite(Hop sharedInput) {
138140 }
139141
140142 int consumerCount = sharedInput .getParent ().size ();
141- if (LOG .isDebugEnabled ()) {
142- LOG .debug ("Inject tee for hop " + sharedInput .getHopID () + " ("
143- + sharedInput .getName () + "), consumers=" + consumerCount );
144- }
143+ LOG .debug ("Inject tee for hop {} ({}), consumers={}" , sharedInput .getHopID (), sharedInput .getName (), consumerCount );
145144
146145 // Take a defensive copy of consumers before modifying the graph
147146 ArrayList <Hop > consumers = new ArrayList <>(sharedInput .getParent ());
@@ -161,10 +160,7 @@ private void applyTopDownTeeRewrite(Hop sharedInput) {
161160 handledHop .put (sharedInput .getHopID (), teeOp );
162161 rewrittenHops .add (sharedInput .getHopID ());
163162
164- if (LOG .isDebugEnabled ()) {
165- LOG .debug ("Created tee hop " + teeOp .getHopID () + " -> "
166- + teeOp .getName ());
167- }
163+ LOG .debug ("Created tee hop {} -> {}" , teeOp .getHopID (), teeOp .getName ());
168164 }
169165
170166 @ SuppressWarnings ("unused" )
@@ -276,11 +272,8 @@ private void removeRedundantTeeChains(Hop hop) {
276272 if (HopRewriteUtils .isData (hop , OpOpData .TEE ) && hop .getInput ().size () == 1 ) {
277273 Hop teeInput = hop .getInput ().get (0 );
278274 if (HopRewriteUtils .isData (teeInput , OpOpData .TEE )) {
279- if (LOG .isDebugEnabled ()) {
280- LOG .debug ("Remove redundant tee hop " + hop .getHopID ()
281- + " (" + hop .getName () + ") -> " + teeInput .getHopID ()
282- + " (" + teeInput .getName () + ")" );
283- }
275+ LOG .debug ("Remove redundant tee hop {} ({}) -> {} ({})" ,
276+ hop .getHopID (), hop .getName (), teeInput .getHopID (), teeInput .getName ());
284277 HopRewriteUtils .rewireAllParentChildReferences (hop , teeInput );
285278 HopRewriteUtils .removeAllChildReferences (hop );
286279 }
0 commit comments