~/jendhub

def Blk.max_rows_upto

bend-tensors@0.0.0.2 · bend_tensors.bend:429

import bend-tensors@0.0.0.2/bend_tensors.bend as BendTensors
import Base

429# each row's largest among its first k columns
430def Blk.max_rows_upto(k: Nat, m: Blk) -> Vec4:
431  match k m:
432    case 1n Blk{a00, a01, a02, a03, a10, a11, a12, a13, a20, a21, a22, a23, a30, a31, a32, a33}:
433      Vec4{a00, a10, a20, a30}
434    case 2n Blk{a00, a01, a02, a03, a10, a11, a12, a13, a20, a21, a22, a23, a30, a31, a32, a33}:
435      Vec4{F32.max(a00, a01), F32.max(a10, a11), F32.max(a20, a21), F32.max(a30, a31)}
436    case 3n Blk{a00, a01, a02, a03, a10, a11, a12, a13, a20, a21, a22, a23, a30, a31, a32, a33}:
437      Vec4{F32.max(F32.max(a00, a01), a02), F32.max(F32.max(a10, a11), a12), F32.max(F32.max(a20, a21), a22), F32.max(F32.max(a30, a31), a32)}
438    case _ m2:
439      Blk.max_rows(m2)

Uses (5)

Used by (1)