[SYSTEMML-707] Fix rewrite 'simplify empty reorg' (diagv2m dims check) Project: http://git-wip-us.apache.org/repos/asf/incubator-systemml/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-systemml/commit/0dfa71d3 Tree: http://git-wip-us.apache.org/repos/asf/incubator-systemml/tree/0dfa71d3 Diff: http://git-wip-us.apache.org/repos/asf/incubator-systemml/diff/0dfa71d3
Branch: refs/heads/master Commit: 0dfa71d354d3534f087e804b42c88d27e9254a98 Parents: 2c23607 Author: Matthias Boehm <[email protected]> Authored: Fri Jun 3 00:09:26 2016 -0700 Committer: Matthias Boehm <[email protected]> Committed: Fri Jun 3 00:09:26 2016 -0700 ---------------------------------------------------------------------- .../hops/rewrite/RewriteAlgebraicSimplificationDynamic.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-systemml/blob/0dfa71d3/src/main/java/org/apache/sysml/hops/rewrite/RewriteAlgebraicSimplificationDynamic.java ---------------------------------------------------------------------- diff --git a/src/main/java/org/apache/sysml/hops/rewrite/RewriteAlgebraicSimplificationDynamic.java b/src/main/java/org/apache/sysml/hops/rewrite/RewriteAlgebraicSimplificationDynamic.java index 817c839..7003afa 100644 --- a/src/main/java/org/apache/sysml/hops/rewrite/RewriteAlgebraicSimplificationDynamic.java +++ b/src/main/java/org/apache/sysml/hops/rewrite/RewriteAlgebraicSimplificationDynamic.java @@ -938,8 +938,8 @@ public class RewriteAlgebraicSimplificationDynamic extends HopRewriteRule else if( rhi.getOp() == ReOrgOp.REV ) hnew = HopRewriteUtils.createDataGenOp(input, 0); else if( rhi.getOp() == ReOrgOp.DIAG ) { - if( HopRewriteUtils.isDimsKnown(input) ){ - if( input.getDim1()==1 ) //diagv2m + if( HopRewriteUtils.isDimsKnown(input) ) { + if( input.getDim2()==1 ) //diagv2m hnew = HopRewriteUtils.createDataGenOp(input, false, input, true, 0); else //diagm2v hnew = HopRewriteUtils.createDataGenOpByVal(
