[mlpack-git] master: changed size_t to int int binarize to support openmp (55567aa)

gitdub at mlpack.org gitdub at mlpack.org
Wed Jun 15 05:46:14 EDT 2016


Repository : https://github.com/mlpack/mlpack
On branch  : master
Link       : https://github.com/mlpack/mlpack/compare/a0b31abe5ff69117645c664dbeac1476dd5e48f7...2da9c5bac14a00145c757b8139c245913b86e034

>---------------------------------------------------------------

commit 55567aaf49dac9951d7f2e52697fdde3aff387f2
Author: Keon Kim <kwk236 at gmail.com>
Date:   Wed Jun 15 18:46:14 2016 +0900

    changed size_t to int int binarize to support openmp


>---------------------------------------------------------------

55567aaf49dac9951d7f2e52697fdde3aff387f2
 src/mlpack/core/data/binarize.hpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/mlpack/core/data/binarize.hpp b/src/mlpack/core/data/binarize.hpp
index 7a47ffd..6a90db6 100644
--- a/src/mlpack/core/data/binarize.hpp
+++ b/src/mlpack/core/data/binarize.hpp
@@ -40,12 +40,12 @@ void Binarize(const arma::Mat<T>& input,
 {
   output.copy_size(input);
 
-  const size_t totalElems = static_cast<size_t>(input.n_elem);
+  const int totalElems = static_cast<int>(input.n_elem);
   const T *inPtr = input.memptr();
   T *outPtr = output.memptr();
 
   #pragma omp parallel for
-  for (size_t i = 0; i < totalElems; ++i)
+  for (int i = 0; i < totalElems; ++i)
   {
     if (inPtr[i] > threshold)
       outPtr[i] = 1;
@@ -84,7 +84,7 @@ void Binarize(const arma::Mat<T>& input,
   output = input;
 
   #pragma omp parallel for
-  for (size_t i = 0; i < input.n_cols; ++i)
+  for (int i = 0; i < input.n_cols; ++i)
   {
     if (input(dimension, i) > threshold)
       output(dimension, i) = 1;




More information about the mlpack-git mailing list