diff options
author | Trevor Killeen <killeentm@gmail.com> | 2017-05-05 23:04:27 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2017-05-09 02:12:10 +0300 |
commit | fdc73fa7bd34ebe3e3eeb6a0b7202fe9bdd93be8 (patch) | |
tree | 37a3c9d3d7e12502623238ddef4f418b967df812 | |
parent | 8fb354fa08cd98eccbff13bbdc99fc2b3e07824f (diff) |
use current stream in cat array kernel launch
-rw-r--r-- | lib/THC/generic/THCTensorMath.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/lib/THC/generic/THCTensorMath.cu b/lib/THC/generic/THCTensorMath.cu index e9f697d..4c609ba 100644 --- a/lib/THC/generic/THCTensorMath.cu +++ b/lib/THC/generic/THCTensorMath.cu @@ -203,7 +203,7 @@ void THCTensor_(catArray)(THCState *state, THCTensor *result, // Template Declarations for dim = 1, 2, 3, 4 #define HANDLE_CASE(DIMS) \ - CatArrayBatchedCopy<real, unsigned int, DIMS><<<applyGrid, applyBlock>>>(data, d_inputs, param, cat_dimension, param.outputStride[cat_dimension]); + CatArrayBatchedCopy<real, unsigned int, DIMS><<<applyGrid, applyBlock, 0, THCState_getCurrentStream(state)>>>(data, d_inputs, param, cat_dimension, param.outputStride[cat_dimension]); // Now we loop offset = 0; |