-
-
Notifications
You must be signed in to change notification settings - Fork 124
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Avoiding scalar indexing in cuda #313
Conversation
The CI failure appears to be unrelated to this change? AFAICT none of the conv machinery uses |
|
Mostly looks good. |
Reopening to see what's up with CI |
Yeah, seems CI is unrelated. @yuehhua happy to merge if you can address replacing this with |
The functionality of this function is to calculate the maximum values across indices. For example, I have a array of tuples that contains a series of indices. julia> A = [(5, 2), (3, 4), (1, 6)];
julia> maximum_dims(dims::AbstractArray{NTuple{N, T}}) where {N,T} = Tuple(maximum(x->x[i], dims) for i = 1:N);
julia> maximum_dims(A)
(5, 6)
julia> maximum(A)
(5, 2) The maximum value of first component across indices is 5 and the maximum value of second component is 6. So, I want |
Co-authored-by: Carlo Lucibello <carlo.lucibello@gmail.com>
Thanks all |
Existing implementation use scalar indexing for cuarray. To avoid using scalar indexing, refactoring is done.