diff options
| author | Soumith Chintala <soumith@gmail.com> | 2015-01-07 09:23:38 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2015-01-07 09:23:38 +0300 |
| commit | 81d2c4215451b350404364dfc19ef5250fe6155b (patch) | |
| tree | 0da8a66d6be4c6e57401f61a9571f2b90004ca4b | |
| parent | 75fd6934fa876dc3ee191fbf29e9ad689af037f4 (diff) | |
openmp fallback for SparseLinear.c optimizations
| -rw-r--r-- | generic/SparseLinear.c | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/generic/SparseLinear.c b/generic/SparseLinear.c index b3ccbf1..ad26641 100644 --- a/generic/SparseLinear.c +++ b/generic/SparseLinear.c @@ -43,7 +43,11 @@ static int nn_(SparseLinear_updateOutput)(lua_State *L) THTensor_(zero)(buffer); #pragma omp parallel for private(i) schedule(static) num_threads(num_shards) for (i = 0; i < input->size[0]; i++) { +#ifdef _OPENMP int shardId = omp_get_thread_num(); +#else + int shardId = 1; +#endif long offset = (long)(THTensor_(get2d)(input, i, 0)) - 1; if (offset >= 0 && offset < inDim) { |