Add comments and fix error message

apache · Dec 5, 2017 · a1b3db0 · a1b3db0
1 parent 362af57
commit a1b3db0
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/benchmark/python/sparse/dot.py b/benchmark/python/sparse/dot.py
@@ -342,7 +342,7 @@ def run_benchmark(ctx=None, lhs="csr", lhs_trans=False, rhs="dns", fw="mxnet", r
                       distribution="uniform"):
 
         if rhs_density > 1 or rhs_density < 0:
-            raise ValueError("Value other than csr for lhs not supported")
+            raise ValueError("rhs_density has to be between 0 and 1")
 
         print_benchmark_info(lhs, rhs, lhs_trans, fw)
 

diff --git a/src/operator/tensor/dot-inl.h b/src/operator/tensor/dot-inl.h
@@ -536,6 +536,8 @@ struct DotCsrTransRspRspByRowBlocks {
 /*!
  * \brief CPU Kernel of PopulateCsrForNNC
  * Parallelization by individual rows
+ * Populates the indptr and indices array
+ * based on number of non zero columns
  */
 struct PopulateCsrForNNC {
   /*!