mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
840e71f4e6
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 |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| group_spatial_softmax_op.cc | ||
| group_spatial_softmax_op.cu | ||
| group_spatial_softmax_op.h | ||
| ps_roi_pool_op.cc | ||
| ps_roi_pool_op.cu | ||
| ps_roi_pool_op.h | ||
| roi_pool_f_op.cc | ||
| roi_pool_f_op.cu | ||
| roi_pool_f_op.h | ||
| sample_as_op.cc | ||
| sample_as_op.cu | ||
| sample_as_op.h | ||
| select_smooth_l1_loss_op.cc | ||
| select_smooth_l1_loss_op.cu | ||
| select_smooth_l1_loss_op.h | ||
| sigmoid_cross_entropy_loss_op.cc | ||
| sigmoid_cross_entropy_loss_op.cu | ||
| sigmoid_cross_entropy_loss_op.h | ||
| sigmoid_focal_loss_op.cc | ||
| sigmoid_focal_loss_op.cu | ||
| sigmoid_focal_loss_op.h | ||
| smooth_l1_loss_op.cc | ||
| smooth_l1_loss_op.cu | ||
| smooth_l1_loss_op.h | ||
| softmax_focal_loss_op.cc | ||
| softmax_focal_loss_op.cu | ||
| softmax_focal_loss_op.h | ||
| spatial_narrow_as_op.cc | ||
| spatial_narrow_as_op.cu | ||
| spatial_narrow_as_op.h | ||
| upsample_nearest_op.cc | ||
| upsample_nearest_op.cu | ||
| upsample_nearest_op.h | ||
| upsample_nearest_op_test.py | ||