Skip to content

Commit

Permalink
Fix the cast and shape issues with Scatter and Gather ops (#174)
Browse files Browse the repository at this point in the history 
- check for numel()==0 for input tensors
- clean up duplicate and redundant code and headers
- move some ops from block list to allow list
  • Loading branch information
@razarmehr @pytorchmergebot
razarmehr authored and pytorchmergebot committed Jan 4, 2023
1 parent 2004df9 commit 88aa616
Show file tree
Hide file tree
Showing 2 changed files with 89 additions and 177 deletions.
Loading

0 comments on commit 88aa616

Please sign in to comment.