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
This might be already documented elsewhere, but when trying to execute a numpy-based code with cunumeric in place of numpy, some of the reductions such arr.sum(axis=(-3, -2, -1)) fails with NotImplementedError: Need support for reducing multiple dimensions.
A simple workaround is to instead do it dimensions by dimension with arr.sum(axis=-1).sum(axis=-1).sum(axis=-1), but it would be nice to have the multi-dimensional reduction. Having this limitation mentioned in the documentation of cunumeric.sum could also be useful.
The text was updated successfully, but these errors were encountered:
CharlelieLrt
changed the title
Sum on 3 dimensions raises
Sum on 3 dimensions raises NotImplementedError
Jan 2, 2024
This might be already documented elsewhere, but when trying to execute a numpy-based code with cunumeric in place of numpy, some of the reductions such
arr.sum(axis=(-3, -2, -1))
fails withNotImplementedError: Need support for reducing multiple dimensions
.A simple workaround is to instead do it dimensions by dimension with
arr.sum(axis=-1).sum(axis=-1).sum(axis=-1)
, but it would be nice to have the multi-dimensional reduction. Having this limitation mentioned in the documentation ofcunumeric.sum
could also be useful.The text was updated successfully, but these errors were encountered: