Tensorization using a library

Is there an example of tensorization, where intrinsic function is a call to (BLAS)-like library?

Are you planning on using a GPU BLAS library or a CPU BLAS library? My guess is that the CPU case will be somewhat easier, as there is not as much boilerplate code e.g., for initializing the device and copying memory that there would be in the GPU case.

Thanks for the reply!
I was looking at a CPU BLAS library written in C/C++. Any pointers to code which does something similar would be very helpful.

Can you see if these examples help you?
https://docs.tvm.ai/tutorials/language/tensorize.html
https://docs.tvm.ai/tutorials/language/intrin_math.html

1 Like

Thanks! I am going through these