from algorithm import vectorize
from tensor import Tensor
alias type = DType.int64
alias nelts = simdwidthof[type]()
def main():
let size = 10
var a = Tensor[type](size)
var b = Tensor[type](size)
var c = Tensor[type](size)
for i in range(size):
a[i] = i + 1
b[i] = i
@parameter
fn diff[nelts : Int](x : Int):
#print(a.simd_load[nelts](x))
c.simd_store[nelts](x, a.simd_load[nelts](x) - b.simd_load[nelts](x))
#print(nelts, x)
vectorize[nelts, diff](size)
for i in range(size):
print(c[i])
#print(a.simd_load[nelts](0))
from algorithm import vectorize
from tensor import Tensor
alias type = DType.int64
alias nelts = simdwidthof[type]()
def main():
let size = 10
var a = Tensor[type](size)
var b = Tensor[type](size)
var c = Tensor[type](size)
for i in range(size):
a[i] = i + 1
b[i] = i
@parameter
fn diff[nelts : Int](x : Int):
#print(a.simd_load[nelts](x))
c.simd_store[nelts](x, a.simd_load[nelts](x) - b.simd_load[nelts](x))
#print(nelts, x)
vectorize[nelts, diff](size)
for i in range(size):
print(c[i])
#print(a.simd_load[nelts](0))