diff options
author | Soumith Chintala <soumith@gmail.com> | 2014-12-27 21:08:10 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2014-12-27 21:08:10 +0300 |
commit | b071b6afcc4d67f9b0b32c075dfa1bab601ce17a (patch) | |
tree | 38f1f513840555161520882b1e26b0e9fea01a34 | |
parent | 6069bc49a8b5d3ccb5d2831355b16457e3bc8d40 (diff) |
removing omp parallel in VolumetricConvolution batch mode, because of segfaults over multiple threads (on some compilers)
-rw-r--r-- | generic/VolumetricConvolution.c | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/generic/VolumetricConvolution.c b/generic/VolumetricConvolution.c index a7960dc..bb30a70 100644 --- a/generic/VolumetricConvolution.c +++ b/generic/VolumetricConvolution.c @@ -55,7 +55,6 @@ static int nn_(VolumetricConvolution_updateOutput)(lua_State *L) THTensor *inb = THTensor_(new)(); THTensor *outb = THTensor_(new)(); -#pragma omp parallel for private(j) for (j=0; j<nBatch; j++) { /* loop over batches */ THTensor_(select)(inb,input,0,j); THTensor_(select)(outb,output,0,j); @@ -113,7 +112,6 @@ static int nn_(VolumetricConvolution_updateGradInput)(lua_State *L) THTensor_(resize5d)(gradInput, input->size[0], input->size[1], input->size[2], input->size[3], input->size[4]); -#pragma omp parallel for private(j) for (j=0; j<nBatch; j++) { /* loop over batches */ THTensor_(select)(ginpb,gradInput,0,j); THTensor_(select)(goutb,gradOutput,0,j); @@ -172,7 +170,6 @@ static int nn_(VolumetricConvolution_accGradParameters)(lua_State *L) THTensor *goutb = THTensor_(new)(); long j; -#pragma omp parallel for private(j) for (j=0; j<nBatch; j++) { /* loop over batches */ THTensor_(select)(inpb,input,0,j); THTensor_(select)(goutb,gradOutput,0,j); |