I have submitted a PR to add support for
axis keyword argument to
np.argmax: [WIP] Add preliminary support for axis to argmax by itamarst · Pull Request #6928 · numba/numba · GitHub
The functionality appears to work, but I’ve… broken parallelization, and I have no idea how re-enable it. https://dev.azure.com/numba/numba/_build/results?buildId=8489&view=logs&j=f0d6d692-dedb-59c9-30d1-aa1d6b8347b2&t=6ceabc93-1c05-51c3-a86f-8b00af8679b9 shows the failing test.
I’m not really sure how to reenable parallelization (was switching to
@overload a mistake? I couldn’t get
@lower_builtin to work), so any advice would be welcome.