BendMlLlama.depth_sel: def of bend-ml-llama@0.2.1.0
Import: import bend-ml-llama@0.2.1.0/main.bend as BendMlLlama
Lines 1458-1466 of bend-ml-llama@0.2.1.0/main.bend: https://jend.leeeet.dev/read.txt?path=bend-ml-llama@0.2.1.0/main.bend&start_line=1458&end_line=1466

import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

# the weight matrices are kept in 2^d row bands (TA.Bands); each matrix · vector product runs its bands in
# parallel without copying the weights (bend-ml NOTES exp. 11). With 2 threads or fewer the bands cannot help,
# so d = 0: a single band, which computes what the sequential Mat.matmul_nt computes (bend-ml exp. 14).
def depth_sel(few: Bool) -> Nat:
  match few:
    case True{}:
      0n
    case False{}:
      4n

Uses (2):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Bool
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat