Skip to content

Commit bfb49d9

Browse files
author
Joe Petviashvili
committed
fix types for sum of booleans and bytes
1 parent 7acc16d commit bfb49d9

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

src/util.jl

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -492,3 +492,15 @@ function sortbykey{T1,T2,N}(keys::AFArray{T1,N},values::AFArray{T2,N},dim::Integ
492492
_error(ccall((:af_sort_by_key,af_lib),af_err,(Ptr{af_array},Ptr{af_array},af_array,af_array,UInt32,Bool),out_keys,out_values,keys.arr,values.arr,UInt32(dim - 1),isAscending))
493493
(AFArray{T1,N}(out_keys[]),AFArray{T2,N}(out_values[]))
494494
end
495+
496+
function sum{N}(_in::AFArray{UInt8,N},dim::Integer)
497+
out = RefValue{af_array}(0)
498+
_error(ccall((:af_sum,af_lib),af_err,(Ptr{af_array},af_array,Cint),out,_in.arr,Cint(dim - 1)))
499+
AFArray{UInt32,N}(out[])
500+
end
501+
502+
function sum{N}(_in::AFArray{Bool,N},dim::Integer)
503+
out = RefValue{af_array}(0)
504+
_error(ccall((:af_sum,af_lib),af_err,(Ptr{af_array},af_array,Cint),out,_in.arr,Cint(dim - 1)))
505+
AFArray{UInt32,N}(out[])
506+
end

0 commit comments

Comments
 (0)