pytorch
ceb16d88 - [Bootcamp] add CUDA kernel checks to ATen/native/cuda (#47466)

Commit
4 years ago
[Bootcamp] add CUDA kernel checks to ATen/native/cuda (#47466) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/47466 - Add kernel launch check `TORCH_CUDA_KERNEL_LAUNCH_CHECK()` (D24309971 (https://github.com/pytorch/pytorch/commit/353e7f940f548e0a0cb3b420b4190b4624ae9b41)) to several files in aten/src/ATen/native/cuda - Get rid of old check `AT_CUDA_CHECK(cudaGetLastError())` in these same files Test Plan: Test build: ``` buck build //caffe2/aten:ATen-cu ``` To check for launches without checks: ``` python3 caffe2/torch/testing/check_kernel_launches.py ``` Make sure none of the updated files are in the returned list: {F343234608} Reviewed By: r-barnes Differential Revision: D24724947 fbshipit-source-id: a7c7d3c70ed8fb5dfd69997b50f9c838f8651791
Author
Cameron Burnett
Parents
Loading