Yixin Bao 840e71f4e6 Check CUDA kernel launches (/fbcode/caffe2/) (#49145) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/49145 Pull Request resolved: https://github.com/pytorch/pytorch/pull/49105 (1) Add a safety check `C10_CUDA_KERNEL_LAUNCH_CHECK()` after each kernel launch. This diff only changes the files inside the directory /fbsource/fbcode/caffe2/modules/, /fbsource/fbcode/caffe2/fb/, /fbsource/fbcode/caffe2/test/. (2) Get rid of old check `AT_CUDA_CHECK(cudaGetLastError())` when necessary. Test Plan: Test build: ``` buck build mode/dev-nosan //caffe2/modules/detectron: buck test mode/dev-nosan //caffe2/modules/detectron: buck build mode/dev-nosan //caffe2/torch/fb/: buck test mode/dev-nosan //caffe2/torch/fb/: ``` 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. Reviewed By: r-barnes Differential Revision: D25452852 fbshipit-source-id: d6657edab612c9e0fa99b29c68460be8b1a20064		2020-12-10 10:43:03 -08:00
..
no_python_abi_suffix_test
self_compiler_include_dirs_test
torch_test_cpp_extension	remediation of S205607	2020-07-17 17:19:47 -07:00
cpp_c10d_extension.cpp	[c10d] switch ProcessGroup to be managed by intrusive_ptr (#47343)	2020-11-12 07:36:23 -08:00
cpp_c10d_extension.hpp	[c10d] switch ProcessGroup to be managed by intrusive_ptr (#47343)	2020-11-12 07:36:23 -08:00
cpp_frontend_extension.cpp
cuda_extension.cpp
cuda_extension.cu	Check CUDA kernel launches (/fbcode/caffe2/) (#49145)	2020-12-10 10:43:03 -08:00
cuda_extension_kernel.cu	Check CUDA kernel launches (/fbcode/caffe2/) (#49145)	2020-12-10 10:43:03 -08:00
cuda_extension_kernel2.cu	Check CUDA kernel launches (/fbcode/caffe2/) (#49145)	2020-12-10 10:43:03 -08:00
cudnn_extension.cpp
doubler.h
extension.cpp
jit_extension.cpp
jit_extension2.cpp
msnpu_extension.cpp	Make empty c10-full (#46092)	2020-11-12 17:08:21 -08:00
rng_extension.cpp
setup.py	Hipify revamp [REDUX] (#48715)	2020-12-02 18:03:23 -08:00
torch_library.cu	Test TORCH_LIBRARY in CUDA extension (#47524)	2020-11-16 13:12:22 -08:00