Unverified Commit 170160a5 authored by Yuxin Wu's avatar Yuxin Wu Committed by GitHub
Browse files

Raise kernel launch errors instead of just print error message in cuda ops (#7080)


Co-authored-by: default avatarYuxin Wu <ppwwyyxx@users.noreply.github.com>
parent f84af6d1
......@@ -300,11 +300,7 @@ void deformable_im2col(
data_col.data_ptr<scalar_t>());
}));
}
cudaError_t err = cudaGetLastError();
if (err != cudaSuccess) {
printf("error in deformable_im2col: %s\n", cudaGetErrorString(err));
}
C10_CUDA_KERNEL_LAUNCH_CHECK();
}
int get_greatest_divisor_below_bound(int n, int bound) {
......@@ -483,11 +479,7 @@ void compute_grad_input(
grad_im.data_ptr<scalar_t>());
}));
}
cudaError_t err = cudaGetLastError();
if (err != cudaSuccess) {
printf("error in compute_grad_input: %s\n", cudaGetErrorString(err));
}
C10_CUDA_KERNEL_LAUNCH_CHECK();
}
template <typename scalar_t, typename index_t>
......@@ -736,12 +728,7 @@ void compute_grad_offset_and_mask(
grad_mask.data_ptr<scalar_t>());
}));
}
cudaError_t err = cudaGetLastError();
if (err != cudaSuccess) {
printf(
"error in compute_grad_offset_and_mask: %s\n", cudaGetErrorString(err));
}
C10_CUDA_KERNEL_LAUNCH_CHECK();
}
std::tuple<at::Tensor, at::Tensor, at::Tensor> backward_gradient_inputs(
......
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