Skip to content

GitLab

Commit 53c52678 authored by Daniel Povey's avatar Daniel Povey
Browse files

Remove print statement'

parent 1dc0d559
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
torch_learned_nonlin/learned_nonlin_cuda_kernel.cu
View file @ 53c52678
...@@ -146,7 +146,6 @@ void learned_nonlin_kernel( ...@@ -146,7 +146,6 @@ void learned_nonlin_kernel(
for (int i = 0; i < K; i++) { for (int i = 0; i < K; i++) {
int isign = i * sign; int isign = i * sign;
y_vals[K + isign] = sum * scale; y_vals[K + isign] = sum * scale;
printf("c = %d, y_vals[%d] = %f\n", c, K + isign, sum * scale);
sum += params_buf[Koffset + isign]; sum += params_buf[Koffset + isign];
} }
y_vals[0] = y_vals[1]; // Both threads do this but it's OK. y_vals[0] = y_vals[1]; // Both threads do this but it's OK.
...@@ -584,10 +583,11 @@ torch::Tensor learned_nonlin_cuda(torch::Tensor input, ...@@ -584,10 +583,11 @@ torch::Tensor learned_nonlin_cuda(torch::Tensor input,
int shared_mem_numel = 2 * N + 3; int shared_mem_numel = 2 * N + 3;
std::cout << "C,B,T,N = " << C << "," << B << "," << T << "," << N if (false)
<< ", images_per_thread_block = " << images_per_thread_block std::cout << "C,B,T,N = " << C << "," << B << "," << T << "," << N
<< ", grid_dim_y = " << grid_dim_y << ", images_per_thread_block = " << images_per_thread_block
<< "\n"; << ", grid_dim_y = " << grid_dim_y
<< "\n";
TORCH_CHECK(THREADS_PER_BLOCK / images_per_thread_block >= T || TORCH_CHECK(THREADS_PER_BLOCK / images_per_thread_block >= T ||
images_per_thread_block == 1, images_per_thread_block == 1,
......
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