Does TASO support ARM CPUs

jiazhihao / TASO

The Tensor Algebra SuperOptimizer for Deep Learning

Apache License 2.0

682 stars 89 forks source link

Does TASO support ARM CPUs #71

Open hashimSharif opened 3 years ago

hashimSharif commented 3 years ago

Hi, TASO docker container seems to use nvidia-docker and the paper talks about experiments on Nvidia V100. I am wondering if TASO can be used for optimizing models on a Raspberry Pi4 with a 64 bit ARM CPU?

Help is appreciated

ryujaehun commented 2 years ago

In my case, i was working without docker in nvidia xavier.