extend("gpu")

a = [
  [1, 2],
  [3, 4]
]

b = [
  [5, 6],
  [7, 8]
]

// Convert to tensor (Float32)
ta = gpu:to_tensor(a)
tb = gpu:to_tensor(b)

// Multiply on GPU
tc = gpu:multiply(ta, tb)

// Convert result back to array for inspection
result = gpu:to_array(tc)

slog(result)

