def St.plan_dtype

bendtorch-load@1.3.0.0 · main.bend:471

import Base

471def St.plan_dtype(f32: Bool, bf16: Bool, +name: String, +dt: String, +start: W64, +sh: List<&2, W64>, +want: List<&2, U32>, of: List<&2, W64>) -> Plan:
472  match f32 bf16:
473    case True{} _: St.plan_shape(St.shape_eq(sh, want), name, True{}, start, sh, want, of)
474    case _ True{}: St.plan_shape(St.shape_eq(sh, want), name, False{}, start, sh, want, of)
475    case _ _: PlanBad{"tensor " ++ name ++ ": dtype " ++ dt ++ " (only F32 and BF16 are read)"}