FBGEMM
FBGEMM copied to clipboard
Pipelined-hip embeddingBag forward.
The credits of the pipelined-hip implementation belong to @carlushuang.
Deploy Preview for eclectic-stroopwafel-199537 canceled.
| Name | Link |
|---|---|
| Latest commit | 8c38e2bcb8e1b4745c6ec6a61ea778da1aced27d |
| Latest deploy log | https://app.netlify.com/sites/eclectic-stroopwafel-199537/deploys/63101076edf3e000097442bd |
Note, we keep AMD optimal kernel for embeddings as separate module (fbgemm_gpu/hip_kernel) is to facilitate:
- vendor innovation with a degree of freedom to drop-in the very best and very latest
- the kernels there will be HIP native, no need to hipify different than code in codegen
- functionality and performance prove Acknowledge that kernel code here is yet to be beautified, coding style and naming convention are neither similar.