def t8_comb

bend-ml-tensor-array@0.1.19.0 · main.bend:950

import Base

950# dotr's final sum after k products: the sums sit rotated by k mod 4 when its loop ends
951def t8_comb(+rot: Nat, +s0: F32, +s1: F32, +s2: F32, +s3: F32) -> F32:
952  match rot:
953    case 0n:
954      ((s0 + s1 : F32) + (s2 + s3 : F32) : F32)
955    case 1n:
956      ((s1 + s2 : F32) + (s3 + s0 : F32) : F32)
957    case 2n:
958      ((s2 + s3 : F32) + (s0 + s1 : F32) : F32)
959    case _:
960      ((s3 + s0 : F32) + (s1 + s2 : F32) : F32)