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);
        }
        

Reply via email to