def place

stelliferous@0.0.2.0 · tensor_layout.bend:246

import Base
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/tensor_view.bend as Views
import stelliferous@0.0.2.0/tensor_operations.bend as Operations
import stelliferous@0.0.2.0/tensor.bend as Tensors

246# The part written through a view of the target storage: one checked copy,
247# which rejects a view whose extents differ from the part. A part with no
248# element writes nothing; its extents are compared here.
249def place(~Element: Data,name: String,empty: Bool,slice: Maybe<&2,Views.View>,source: Maybe<&2,Views.View>,
250  target: Storage.Buffer<Element>,+target_shape: List<&2,U32>,target_fill: Element,
251  part: Storage.Buffer<Element>,+part_shape: List<&2,U32>,part_fill: Element,+workers: U32) -> Tensors.Tensor<Element> & Tensors.Tensor<Element>:
252  match empty slice source:
253    case False{} Some{slice} Some{source}:
254      placed(~Element,name,target_shape,part_shape,target_fill,part_fill,workers,settle(~Element,True{},Operations.scatter(~Element,part,source,target,slice)))
255    case True{} Some{slice} source:
256      settled(~Element,name,same_extents(slice,part_shape),part,target,target_shape,part_shape,target_fill,part_fill,workers)
257    case empty slice source: settled(~Element,name,False{},part,target,target_shape,part_shape,target_fill,part_fill,workers)