Fixing bug in sparse matrix buffer estimation

henrykironde · Aug 12, 2016 · 66e4534 · 66e4534
1 parent 59fd877
commit 66e4534
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Source/Math/GPUSparseMatrix.h b/Source/Math/GPUSparseMatrix.h
@@ -165,9 +165,9 @@ class MATH_API GPUSparseMatrix : public BaseMatrix<ElemType>
         else if (format == matrixFormatSparseBlockRow)
             return (bufferSize - 2 * sizeof(GPUSPARSE_INDEX_TYPE) * numRows) / sizeof(ElemType);
         else if (format == matrixFormatSparseCSC)
-            return (bufferSize - sizeof(GPUSPARSE_INDEX_TYPE) * (numCols + 1)) / (2 * sizeof(ElemType));
+            return (bufferSize - sizeof(GPUSPARSE_INDEX_TYPE) * (numCols + 1)) / (sizeof(GPUSPARSE_INDEX_TYPE) + sizeof(ElemType));
         else if (format == matrixFormatSparseCSR)
-            return (bufferSize - sizeof(GPUSPARSE_INDEX_TYPE) * (numRows + 1)) / (2 * sizeof(ElemType));
+            return (bufferSize - sizeof(GPUSPARSE_INDEX_TYPE) * (numRows + 1)) / (sizeof(GPUSPARSE_INDEX_TYPE) + sizeof(ElemType));
         else
             NOT_IMPLEMENTED;
     }