Skip to content

GitLab

Commit 201922a0 authored by rusty1s's avatar rusty1s
Browse files

remove prints

parent 2693efc9
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 6 deletions
csrc/cuda/segment_coo_cuda.cu
View file @ 201922a0
...@@ -215,12 +215,6 @@ segment_coo_cuda(torch::Tensor src, torch::Tensor index, ...@@ -215,12 +215,6 @@ segment_coo_cuda(torch::Tensor src, torch::Tensor index,
auto N = out.size(dim); auto N = out.size(dim);
auto avg_len = (float)E_2 / (float)N; auto avg_len = (float)E_2 / (float)N;
std::cout << "E " << E << std::endl;
std::cout << "E2 " << E_2 << std::endl;
std::cout << "E1 " << E_1 << std::endl;
std::cout << "K " << K << std::endl;
std::cout << "N " << N << std::endl;
auto index_info = at::cuda::detail::getTensorInfo<int64_t, int>(index); auto index_info = at::cuda::detail::getTensorInfo<int64_t, int>(index);
auto stream = at::cuda::getCurrentCUDAStream(); auto stream = at::cuda::getCurrentCUDAStream();
AT_DISPATCH_ALL_TYPES(src.scalar_type(), "segment_coo_kernel", [&] { AT_DISPATCH_ALL_TYPES(src.scalar_type(), "segment_coo_kernel", [&] {
......
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