[Bug]: Error in the function nonzero for certain inputs
Opened this issue · 0 comments
Hakdag97 commented
What happened?
For certain inputs (i.e., if only one element in the input list is True), the function nonzero generates an error concerning the redistribution.
Code snippet triggering the error
No response
Error message or erroneous outcome
No response
Version
main (development branch)
Python version
None
PyTorch version
None
MPI version
No response