 #define HANDLE_CASE(TYPE, A, B, C)                                      \
   THCudaTensor_pointwiseApply3RTC(aInfo, bInfo, cInfo,			\
       apply_header, op_string, (TYPE)totalElements, grid, block,	\
-      A, B, C);								\
+      A, B, C, stream);								\
 
 #define HANDLE_C_CASE(TYPE, A, B, C)             \
   {                                              \

                                  const char* op_string)
{
  TensorArgType aType = ReadWrite;
  long totalElements = THCudaTensor_nElement(state, a);

  if (THCudaTensor_nDimension(state, a) > MAX_CUTORCH_DIMS) {

  compilePTX(src, headers, includeNames, ptx);

  void *args[] = {(void*)&aInfo, (void*)&totalElements};
  launch(ptx.data(), "kernel", args, grid, block, (CUstream)stream);
}

// Example op: 'x = x*y'
