free bernoulli tensor

rusty1s · rusty1s · commit d6de3071f83e · 2018-05-17T15:25:14.000+02:00
diff --git 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;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -30,7 +30,9 @@ int THCudaLongTensor_color(THCState state, THCudaLongTensor self) {`
`30`	`30`
`31`	`31`	`KERNEL_RUN(colorKernel, nNodes, selfData, bernoulliData);`
`32`	`32`
	`33`	`+ THCudaByteTensor_free(state, bernoulli);`
`33`	`34`	`int done; cudaMemcpyFromSymbol(&done, d_done, sizeof(done), 0, cudaMemcpyDeviceToHost);`
	`35`	`+`
`34`	`36`	`return done;`
`35`	`37`	`}`
`36`	`38`