def gat.one
import bend-parallel@0.1.0.0/sort.bend as Sortimport Base
423# dst[j] <- src[x].
424def gat.one(~T: Data, +j: U32, +x: U32, p: Sd<T>) -> Sd<T>:
425 match p:
426 case Sd{src, dst}:
427 gat.at(~T, j, dst, Array.get(T, src, x))
import bend-parallel@0.1.0.0/sort.bend as Sortimport Base
423# dst[j] <- src[x].
424def gat.one(~T: Data, +j: U32, +x: U32, p: Sd<T>) -> Sd<T>:
425 match p:
426 case Sd{src, dst}:
427 gat.at(~T, j, dst, Array.get(T, src, x))