[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