reduction_op with axis + keepdim #845
Labels
bug
Something isn't working
redistribution
Related to distributed tensors
reduction operators
Operators for calculations that reduce tensor dimension(s)
Description
A clear and concise description of the bug and the associated functionality.
reduce_op does not work when the axis argument given to a reduce_op function is smaller than the DNDarrays split axis by one.
To Reproduce
Steps to reproduce the behavior:
_operations.reduce_op
and functions calling it (sum
,max
,min
...) respectively.axis == split-1
+keepdim == True
Two processes:
Hangs on more processes.
Expected behavior
A clear and concise description of what you expected to happen.
Return DNDarray
Illustrative
If applicable, add screenshots or minimal examples to help explain your problem.
Version Info
Which version are you using?
master
Additional comments
Any other comments here.
The text was updated successfully, but these errors were encountered: