commit 066fbcd014fa4092152b2cd04ad1d92fc8d7bd59
author Trevor Killeen <killeentm@gmail.com> Fri May 05 13:04:27 2017 -0700
committer Soumith Chintala <soumith@gmail.com> Mon May 08 16:12:10 2017 -0700
tree 727c57ebd5b27c8f0ad06c4165dcbce83bbb69f8
parent 22bbd7ac33ba51469cc913cb01fcd3b70a42e528

use current stream in cat array kernel launch

generic/THCTensorMath.cu

diff --git a/generic/THCTensorMath.cu b/generic/THCTensorMath.cu
index e9f697d..4c609ba 100644
--- a/generic/THCTensorMath.cu
+++ b/generic/THCTensorMath.cu
@@ -203,7 +203,7 @@
 
     // 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;