Abstract

Irregular segmented operations, like scans and reductions.

Synopsis

val segmented_scan [n] 't : (op: t -> t -> t) -> (ne: t) -> (flags: [n]bool) -> (as: [n]t) -> [n]t
val segmented_reduce [n] 't : (op: t -> t -> t) -> (ne: t) -> (flags: [n]bool) -> (as: [n]t) -> ?[d₃₄].*[d₃₄]t
val replicated_iota [n] : (reps: [n]i64) -> []i64
val segmented_iota [n] : (flags: [n]bool) -> [n]i64
val expand 'a 'b [d₀] : (sz: a -> i64) -> (get: a -> i64 -> b) -> (arr: [d₀]a) -> []b
val expand_reduce 'a 'b [d₀] : (sz: a -> i64) -> (get: a -> i64 -> b) -> (op: b -> b -> b) -> (ne: b) -> (arr: [d₀]a) -> []b
val expand_outer_reduce 'a 'b [n] : (sz: a -> i64) -> (get: a -> i64 -> b) -> (op: b -> b -> b) -> (ne: b) -> (arr: [n]a) -> [n]b

Description

val segmented_scan [n] 't: (op: t -> t -> t) -> (ne: t) -> (flags: [n]bool) -> (as: [n]t) -> [n]t

Segmented scan. Given a binary associative operator op with neutral element ne, computes the inclusive prefix scan of the segments of as specified by the flags array, where true starts a segment and false continues a segment.

val segmented_reduce [n] 't: (op: t -> t -> t) -> (ne: t) -> (flags: [n]bool) -> (as: [n]t) -> ?[d₃₄].*[d₃₄]t

Segmented reduction. Given a binary associative operator op with neutral element ne, computes the reduction of the segments of as specified by the flags array, where true starts a segment and false continues a segment. One value is returned per segment.

val replicated_iota [n]: (reps: [n]i64) -> []i64

Replicated iota. Given a repetition array, the function returns an array with each index (starting from 0) repeated according to the repetition array. As an example, replicated_iota [2,3,1] returns the array [0,0,1,1,1,2].

val segmented_iota [n]: (flags: [n]bool) -> [n]i64

Segmented iota. Given a flags array, the function returns an array of index sequences, each of which is reset according to the flags array. As an examples, segmented_iota [false,false,false,true,false,false,false] returns the array [0,1,2,0,1,2,3].

val expand 'a 'b [d₀]: (sz: a -> i64) -> (get: a -> i64 -> b) -> (arr: [d₀]a) -> []b

Generic expansion function. The function expands a source array into a target array given (1) a function that determines, for each source element, how many target elements it expands to and (2) a function that computes a particular target element based on a source element and the target element number associated with the source. As an example, the expression expand (\x->x) (*) [2,3,1] returns the array [0,2,0,3,6,0].

val expand_reduce 'a 'b [d₀]: (sz: a -> i64) -> (get: a -> i64 -> b) -> (op: b -> b -> b) -> (ne: b) -> (arr: [d₀]a) -> []b

Expansion function equivalent to performing a segmented reduction to the result of a general expansion with a flags vector expressing the beginning of the expanded segments. The function makes use of the intermediate flags vector generated as part of the expansion and the expand_reduce function is therefore more efficient than if a segmented reduction (with an appropriate flags vector) is explicitly followed by a call to expand.

val expand_outer_reduce 'a 'b [n]: (sz: a -> i64) -> (get: a -> i64 -> b) -> (op: b -> b -> b) -> (ne: b) -> (arr: [n]a) -> [n]b

Expansion followed by an ''outer segmented reduce'' that ensures that each element in the result array corresponds to expanding and reducing the corresponding element in the source array.