You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I recently needed this functionality and have started implementing it. If this issue is still available, I’d be happy to continue by adding a portable kernel for aten::_embedding_bag.out in kernels/portable/cpu/op_embedding_bag.cpp, along with the required registration and tests.
@yenhao-huang can you create a branch from the repo pytorch/executorch and create a PR there? There are some CI jobs can only be run from the branch of the repo.
Activity
[-]Implement portable kernel: _embedding_bag[/-][+]Implement portable kernel: _embedding_bag.out[/+]yenhao-huang commentedon Oct 12, 2026
Hi @manuelcandales and @larryliu0820, I’m interested in working on this issue.
I recently needed this functionality and have started implementing it. If this issue is still available, I’d be happy to continue by adding a portable kernel for
aten::_embedding_bag.outinkernels/portable/cpu/op_embedding_bag.cpp, along with the required registration and tests.larryliu0820 commentedon Oct 14, 2026
@yenhao-huang can you create a branch from the repo pytorch/executorch and create a PR there? There are some CI jobs can only be run from the branch of the repo.
yenhao-huang commentedon Oct 14, 2026
Sure, I’ve opened PR #23871. Please let me know if you'd like any changes.