pytorch / FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Other
1.17k stars 474 forks source link

Refactor CUDA callback functor #2908

Closed sryap closed 1 month ago

sryap commented 1 month ago

Summary: This diff creates a common CUDA callback function, called cuda_callback_func, in the kv_db_utils namespace. It can be used as a callback function for cudaStreamAddCallback.

cuda_callback_func casts the input functor into a void function, invokes it and then delete it.

Differential Revision: D60348604

netlify[bot] commented 1 month ago

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
Latest commit a0960cdc59957bd6a886d8afafc970e84191d082
Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66a822acc8b19300085b5c82
Deploy Preview https://deploy-preview-2908--pytorch-fbgemm-docs.netlify.app
Preview on mobile
Toggle QR Code...

QR Code

Use your smartphone camera to open QR code link.

To edit notification comments on pull requests, go to your Netlify site configuration.

facebook-github-bot commented 1 month ago

This pull request was exported from Phabricator. Differential Revision: D60348604

facebook-github-bot commented 1 month ago

This pull request was exported from Phabricator. Differential Revision: D60348604

facebook-github-bot commented 1 month ago

This pull request was exported from Phabricator. Differential Revision: D60348604

facebook-github-bot commented 1 month ago

This pull request has been merged in pytorch/FBGEMM@9851079b0aa38031e549c2ed4571909764009b78.