You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@systemml.apache.org by ni...@apache.org on 2017/10/06 02:22:00 UTC
systemml git commit: [SYSTEMML-1793] [MINOR] GPU right indexing bugfix
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 <np...@us.ibm.com>
Authored: Thu Oct 5 19:18:46 2017 -0700
Committer: Niketan Pansare <np...@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);
}