free bernoulli tensor

d6de3071 · rusty1s · 642d548f · d6de3071
Commit d6de3071 authored May 17, 2018 by rusty1s
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

aten/THC/THCColor.cuh aten/THC/THCColor.cuh +2 -0

No files found.
--- a/aten/THC/THCColor.cuh
+++ b/aten/THC/THCColor.cuh
@@ -30,7 +30,9 @@ int THCudaLongTensor_color(THCState *state, THCudaLongTensor *self) {

  KERNEL_RUN(colorKernel, nNodes, selfData, bernoulliData);

+  THCudaByteTensor_free(state, bernoulli);
  int done; cudaMemcpyFromSymbol(&done, d_done, sizeof(done), 0, cudaMemcpyDeviceToHost);
+
  return done;
 }