Skip to content

Commit d077f93

Browse files
committed
Fix rowVec/colVec
1 parent 994bfdb commit d077f93

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/main/java/org/apache/sysds/hops/rewriter/dml/DMLCodeGenerator.java

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -246,10 +246,10 @@ public static String generateDMLVariables(Set<RewriterStatement> vars) {
246246
if (var.isInstruction()) {
247247
if (var.trueInstruction().equals("rowVec")) {
248248
mId = var.getChild(0).getId();
249-
ncol = 1L;
249+
nrow = 1L;
250250
} else if (var.trueInstruction().equals("colVec")) {
251251
mId = var.getChild(0).getId();
252-
nrow = 1L;
252+
ncol = 1L;
253253
} else if (var.trueInstruction().equals("const")) {
254254
sb.append(var.getId());
255255
sb.append(" = matrix(" + var.getChild(1).getLiteral() + ", rows=" + nrow + ", cols=" + ncol + ")\n");

0 commit comments

Comments
 (0)