https://github.com/dmlc/tvm/tree/master/topi/python/topi/x86 The folder name is “x86” but there is no x86-specific code except conv2d files. Are the rest files also specific to x86? Could they be used for arm cpu?
Thanks in advance.
If you just use a tvm target of “cpu,” it should just work, provided you have the rest of the compile toolchain for ARM ready.
But depending on the vector width (e.g., NEON instruction width) of the CPU you are using, the “rasp” schedules might give better performance. I would also make sure that you are using the best llvm target host specification for your ARM CPU (e.g., llvm -target=armv7l-none-linux-gnueabihf for a raspberry pi 3b).