BendtorchGpu.sc_put: def of bendtorch-gpu@1.3.0.0
Import: import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpu
Lines 755-758 of bendtorch-gpu@1.3.0.0/main.bend: https://jend.leeeet.dev/read.txt?path=bendtorch-gpu@1.3.0.0/main.bend&start_line=755&end_line=758

import Base
import bendtorch@1.3.0.0/main.bend as T

# forward 1: P's cell (i, j) = (q's row i · k's row j) * sc, as qrow's dot and pw's product (S3{q, kᵀ, p})
def sc_put(+e: U32, +sc: F32, p: Array<F32>, r: T.D2) -> S3:
  match r:
    case T.D2{q, k, +s}: S3{q, k, Array.set(F32, p, e, (s * sc : F32))}

Uses (7):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Array
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Array.set
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32.mul
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32
  https://jend.leeeet.dev/definition.txt?path=bendtorch-gpu@1.3.0.0/main.bend&name=S3
  https://jend.leeeet.dev/definition.txt?path=bendtorch@1.3.0.0/main.bend&name=D2