Hello everyone,
I encountered some problems and have been confused for few days, so I come here for some helps. I’m trying to look for the c code for some op (eq. matmul), to the best of my knowledge, the op that the models use in python was written in c code, I expected that there will be a file like “matmul.cc” however I can’t find any file that I want.
There are two ways that I’ve tried, the first is directly using the keyword (tvm matmul c code) to search on the internet the second way is to download the whole repo of tvm github, and search the “matmul” by Window’s search function in the repo file directory that I downloaded.
With second way, I only find “batch_matmul.cc” and “combine_parallel_batch_matmul.cc”, however the code that they contained seems doesn’t like what “matrix multiplication” code that I expected (like using some for loop and maybe some c[i][j]=a[i][k]*b[k][j] ). Meanwhile, as far as I know, the batch_matmul is for tensorflow’s matmul. Does it mean that if I use Pytorch as my model’s frame , then when my model call matmul , it will not use the “batch_matmul.cc” file ? If so, then where will the tvm go to do the calculation of matmul?
Thanks for all your reading and helps