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
The issue is that when enforce_sorted=false the unsorted_indices are used to index on the lengths tensor, but the indices are stored on the same device as the input tensor, and lengths is stored on the CPU.