Skip to content
Snippets Groups Projects
Unverified Commit 942cc2de authored by qasfb's avatar qasfb Committed by GitHub
Browse files

Update implementation of scaled_index_add and index_select_cat with fallback 

Add a check for the proper size and type for the optimized scaled_index_add and index_select_cat, and fallback on pytorch implementation otherwise
parent 6a626154
No related branches found
No related tags found
Hide whitespace changes
Inline Side-by-side
Showing
with 20 additions and 2 deletions
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment