Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Revert thread dim for ToTensor operator
Browse files Browse the repository at this point in the history
  • Loading branch information
sandeep-krishnamurthy committed Feb 14, 2019
1 parent 26a0532 commit cf05dd7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/operator/image/image_random.cu
Original file line number Diff line number Diff line change
Expand Up @@ -114,7 +114,7 @@ void ToTensorImplCUDA(mshadow::Stream<gpu> *s,
}

ToTensorCudaKernel<gpu, DType>
<<<blocks, dim3(H, cuda::kMaxThreadsPerBlock / H), 0, stream>>>(input, output,
<<<blocks, dim3(32, 32), 0, stream>>>(input, output,
req, N, H, W, C, normalize_factor);
MSHADOW_CUDA_POST_KERNEL_CHECK(ToTensorCudaKernel);
}
Expand Down

0 comments on commit cf05dd7

Please sign in to comment.