Skip to content

Commit 3988817

Browse files
committed
Bugfix
1 parent 5f9be9e commit 3988817

File tree

5 files changed

+617
-300
lines changed

5 files changed

+617
-300
lines changed

src/main/java/org/apache/sysds/hops/rewrite/HopRewriteUtils.java

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -393,6 +393,30 @@ public static DataGenOp copyDataGenOp( DataGenOp inputGen, double scale, double
393393

394394
return datagen;
395395
}
396+
397+
public static Hop createDataGenOpFomDims( Hop rows, Hop cols, double value ) {
398+
Hop val = new LiteralOp(value);
399+
400+
HashMap<String, Hop> params = new HashMap<>();
401+
params.put(DataExpression.RAND_ROWS, rows);
402+
params.put(DataExpression.RAND_COLS, cols);
403+
params.put(DataExpression.RAND_MIN, val);
404+
params.put(DataExpression.RAND_MAX, val);
405+
params.put(DataExpression.RAND_PDF, new LiteralOp(DataExpression.RAND_PDF_UNIFORM));
406+
params.put(DataExpression.RAND_LAMBDA, new LiteralOp(-1.0));
407+
params.put(DataExpression.RAND_SPARSITY, new LiteralOp(1.0));
408+
params.put(DataExpression.RAND_SEED, new LiteralOp(DataGenOp.UNSPECIFIED_SEED) );
409+
410+
//note internal refresh size information
411+
Hop datagen = new DataGenOp(OpOpDG.RAND, new DataIdentifier("tmp"), params);
412+
datagen.setBlocksize(1000);
413+
//copyLineNumbers(rowInput, datagen);
414+
415+
if( value==0 )
416+
datagen.setNnz(0);
417+
418+
return datagen;
419+
}
396420

397421
public static Hop createDataGenOp( Hop rowInput, Hop colInput, double value )
398422
{

0 commit comments

Comments
 (0)