pytorch / FBGEMM

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

Add benchmark EmbeddingSpMDMNBitBenchmarkOutTypeFloat16 #2901

Closed helloguo closed 1 month ago

helloguo commented 1 month ago

Summary: This diff adds the benchmark EmbeddingSpMDMNBitBenchmarkOutTypeFloat16, to test TBE with output type float16. This diff doesn't change the EmbeddingSpMDMNBitBenchmark.

Differential Revision: D60254038

netlify[bot] commented 1 month ago

Deploy Preview for pytorch-fbgemm-docs ready!

Name Link
Latest commit 8050e77b418e5e8650716b350e0d7a8fc741135c
Latest deploy log https://app.netlify.com/sites/pytorch-fbgemm-docs/deploys/66a2d2c7cfe0680008bb538f
Deploy Preview https://deploy-preview-2901--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: D60254038

facebook-github-bot commented 1 month ago

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

facebook-github-bot commented 1 month ago

This pull request has been merged in pytorch/FBGEMM@0e561d409ab6514d03d01401e43a480b1ae54e9e.