tvm icon indicating copy to clipboard operation
tvm copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Use tensorization for avx2 as well.

Open kimishpatel opened this issue 6 years ago • 1 comments

Use avx2 based tensorization to enable perf improvement on avx2 only machines. Gets on par with fbgemm but on VMs variation prevents strong conclusion.

kimishpatel avatar Sep 25 '19 20:09 kimishpatel

dmlc/tvm has reverted the changes on which this PR depends. Even though facebookexperimental/tvm head is behind it and it should work fine, but I would just suggest we dont merge this yet.

kimishpatel avatar Sep 26 '19 17:09 kimishpatel