-
Notifications
You must be signed in to change notification settings - Fork 10
Open
Description
As the title says, is there a way to support ThreadsX.argmin
and ThreadsX.argmax
?
Sorry I am too new with Julia to be able to help.
The way I see it is to define the argmax reduction
oneargmax((mxi,mx),(i,ai)) = (mx<ai) ? (i,ai) : (mxi,mx)
# but I do not know how to define the generic initial value for the foldl/reduce in ThreadsX
myargmax(f, itr; init=(nothing,-1)) = ThreadsX.mapreduce(identity, oneargmax, zip(itr,Iterators.map(f,itr)); simd = Val(true), init)
Metadata
Metadata
Assignees
Labels
No labels