correct function description

apache · Jul 28, 2017 · 581ce64 · 581ce64
1 parent b20136d
commit 581ce64
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/operator/tensor/cast_storage-inl.cuh b/src/operator/tensor/cast_storage-inl.cuh
@@ -54,7 +54,7 @@ struct MarkRspRowIdxThreadKernel {
 };
 
 /*!
- * \brief Thread kernel for marking non-zero rows of a tensor.
+ * \brief Warp kernel for marking non-zero rows of a tensor.
  * Parallelized by tensor rows: 1 warp/row
  */
 struct MarkRspRowIdxWarpKernel {
@@ -95,7 +95,7 @@ struct MarkRspRowIdxWarpKernel {
 };
 
 /*!
- * \brief Thread kernel for marking non-zero rows of a tensor.
+ * \brief Block kernel for marking non-zero rows of a tensor.
  * Parallelized by tensor rows: 1 threadBlock/row
  */
 struct MarkRspRowIdxBlockKernel {