Modular•4mo ago

Its possible to create SIMD vector at runtime

I am currently working on a vector implementation using SIMD and I am encountering a challenge. Specifically, I am unable to locate an example demonstrating how to initialize an SIMD vector at runtime. Here is the relevant code snippet I am working with:

# exmaple form matrix multiplication

struct Vector[vector_size:Int]:
    var data:DTypePointer[DType.float32]

    fn __init__(inout self):
        self.data = DTypePointer[DType.float32].alloc(vector_size)
        memset_zero(self.data, vector_size)

    fn load[nelts: Int](self, x: Int) -> SIMD[DType.float32, nelts]:
        return self.data.load[width=nelts](x)

    fn store[nelts: Int](self, x: Int, val: SIMD[DType.float32, nelts]):
        return self.data.store[width=nelts](x, val)

    fn __getitem__(self, x: Int) -> Scalar[DType.float32]:
        return self.load[1](x)

    fn __setitem__(self, x: Int, val: Scalar[DType.float32]):
        self.store[1](x, val)

var score = Vector[corpus_size]() # error

# exmaple form matrix multiplication

struct Vector[vector_size:Int]:
    var data:DTypePointer[DType.float32]

    fn __init__(inout self):
        self.data = DTypePointer[DType.float32].alloc(vector_size)
        memset_zero(self.data, vector_size)

    fn load[nelts: Int](self, x: Int) -> SIMD[DType.float32, nelts]:
        return self.data.load[width=nelts](x)

    fn store[nelts: Int](self, x: Int, val: SIMD[DType.float32, nelts]):
        return self.data.store[width=nelts](x, val)

    fn __getitem__(self, x: Int) -> Scalar[DType.float32]:
        return self.load[1](x)

    fn __setitem__(self, x: Int, val: Scalar[DType.float32]):
        self.store[1](x, val)

var score = Vector[corpus_size]() # error

I would appreciate any guidance or examples on how to properly initialize an SIMD vector at runtime. Your assistance in this matter would be highly valued. Thank you for your time and support.

1 Reply

Ehsan M. Kermani (Modular)•4mo ago

What error do you get? Are you initializing inside the main function?

Gaming

Programming

Its possible to create SIMD vector at runtime