BendMlTensorGpu.rng_mix: def of bend-ml-tensor-gpu@0.1.5.1
Import: import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpu
Lines 604-610 of bend-ml-tensor-gpu@0.1.5.1/main.bend: https://jend.leeeet.dev/read.txt?path=bend-ml-tensor-gpu@0.1.5.1/main.bend&start_line=604&end_line=610

import Base

# mlbend-generative's mix32: xor-shift 16, multiply, xor-shift 13, multiply, xor-shift 16, over U32 that wraps
def rng_mix(+z: U32) -> U32:
  +a = U32.xor(z, U32.shrn(z, 16n))
  +b = (a * 2246822519 : U32)
  +c = U32.xor(b, U32.shrn(b, 13n))
  +e = (c * 3266489917 : U32)
  U32.xor(e, U32.shrn(e, 16n))

Uses (4):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32.mul
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32.shrn
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32.xor