Repository: systemml Updated Branches: refs/heads/master 1f3dfbc19 -> 374b0b197
[SYSTEMML-1793] [MINOR] GPU right indexing bugfix - Since GPU only supports sparse_dense rightindexing, updated the memory budget accordingly. Project: http://git-wip-us.apache.org/repos/asf/systemml/repo Commit: http://git-wip-us.apache.org/repos/asf/systemml/commit/374b0b19 Tree: http://git-wip-us.apache.org/repos/asf/systemml/tree/374b0b19 Diff: http://git-wip-us.apache.org/repos/asf/systemml/diff/374b0b19 Branch: refs/heads/master Commit: 374b0b197f9e2fc76b46e0e72f3b3d2594326734 Parents: 1f3dfbc Author: Niketan Pansare <npan...@us.ibm.com> Authored: Thu Oct 5 19:18:46 2017 -0700 Committer: Niketan Pansare <npan...@us.ibm.com> Committed: Thu Oct 5 19:18:46 2017 -0700 ---------------------------------------------------------------------- src/main/java/org/apache/sysml/hops/IndexingOp.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/systemml/blob/374b0b19/src/main/java/org/apache/sysml/hops/IndexingOp.java ---------------------------------------------------------------------- diff --git a/src/main/java/org/apache/sysml/hops/IndexingOp.java b/src/main/java/org/apache/sysml/hops/IndexingOp.java index 2463c7d..fa67618 100644 --- a/src/main/java/org/apache/sysml/hops/IndexingOp.java +++ b/src/main/java/org/apache/sysml/hops/IndexingOp.java @@ -237,7 +237,8 @@ public class IndexingOp extends Hop @Override protected double computeOutputMemEstimate( long dim1, long dim2, long nnz ) { - double sparsity = OptimizerUtils.getSparsity(dim1, dim2, nnz); + // only dense right indexing supported on GPU + double sparsity = isGPUEnabled() ? 1.0 : OptimizerUtils.getSparsity(dim1, dim2, nnz); return OptimizerUtils.estimateSizeExactSparsity(dim1, dim2, sparsity); }
