Skip to content

GitLab

Commit d6de3071 authored by rusty1s's avatar rusty1s
Browse files

free bernoulli tensor

parent 642d548f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
aten/THC/THCColor.cuh
View file @ d6de3071
...@@ -30,7 +30,9 @@ int THCudaLongTensor_color(THCState *state, THCudaLongTensor *self) { ...@@ -30,7 +30,9 @@ int THCudaLongTensor_color(THCState *state, THCudaLongTensor *self) {
KERNEL_RUN(colorKernel, nNodes, selfData, bernoulliData); KERNEL_RUN(colorKernel, nNodes, selfData, bernoulliData);
THCudaByteTensor_free(state, bernoulli);
int done; cudaMemcpyFromSymbol(&done, d_done, sizeof(done), 0, cudaMemcpyDeviceToHost); int done; cudaMemcpyFromSymbol(&done, d_done, sizeof(done), 0, cudaMemcpyDeviceToHost);
return done; return done;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment