saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/test/cpp_extensions
History
Yixin Bao f5e9ffbc27 Check CUDA kernel launches (/fbcode/caffe2/) (#49105) 
Summary:
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 //caffe2/modules/detectron:
buck build //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: D25325039

fbshipit-source-id: 2043d6e63c7d029c35576d3101c18247ffe92f01
2020-12-09 12:34:55 -08:00
..
no_python_abi_suffix_test
self_compiler_include_dirs_test allow user passing relative paths in include_dirs within setuptools.setup (#38264) 2020-05-13 20:00:12 -07:00
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/) (#49105) 2020-12-09 12:34:55 -08:00
cuda_extension_kernel.cu Check CUDA kernel launches (/fbcode/caffe2/) (#49105) 2020-12-09 12:34:55 -08:00
cuda_extension_kernel2.cu Check CUDA kernel launches (/fbcode/caffe2/) (#49105) 2020-12-09 12:34:55 -08:00
cudnn_extension.cpp
doubler.h
extension.cpp allow user passing relative paths in include_dirs within setuptools.setup (#38264) 2020-05-13 20:00:12 -07:00
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 Port /test/cpp_extensions/rng_extension.cpp to new operator registration API (#39459) 2020-06-26 16:12:54 -07:00
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