FBGEMM
FBGEMM copied to clipboard
Replace `C10_CUDA_KERNEL_LAUNCH_CHECK()` in the `KernelLauncher`
Summary:
- Replace
C10_CUDA_KERNEL_LAUNCH_CHECK()in theKernelLauncher, as the latter does not print FILE and LINE
The existing C10_CUDA_KERNEL_LAUNCH_CHECK() implementation does not print the source file and line number when a CUDA kernel launch throws an error, leaving users confused with a context-less message like CUDA error: invalid arguments. This new check is a slimmed re-implementation of the macro with extra context information added to the error (beyond just file and line number) so that we can at least locate the FBGEMM source file or template where the error first surfaces.
Reviewed By: sryap
Differential Revision: D74364031
Deploy Preview for pytorch-fbgemm-docs ready!
| Name | Link |
|---|---|
| Latest commit | c38171531a9d9eb2f591f7deb1ff02e4aa030bbf |
| Latest deploy log | https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/681cfa2163e1ee0008368fd0 |
| Deploy Preview | https://deploy-preview-4097--pytorch-fbgemm-docs.netlify.app |
| Preview on mobile | Toggle QR Code...Use your smartphone camera to open QR code link. |
To edit notification comments on pull requests, go to your Netlify site configuration.
This pull request was exported from Phabricator. Differential Revision: D74364031