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
Support prefetch pipeline in bounds_check_indices (#4312)
Summary:
Pull Request resolved: #4312
X-link: facebookresearch/FBGEMM#1174
Frontend of D72343128
This diff reduces the grid dimension of the bounds_check_indices
kernel when pipeline prefetching is used (in embedding memory
offloading). We need to use the v2 kernel since v1 does not support
grid dimension reduction.
Reviewed By: jwfromm
Differential Revision: D72365505
fbshipit-source-id: 3c57ad3d9c64a0d0f81df70abae1f50b43a1d0fa
0 commit comments