Skip to content
This repository was archived by the owner on Apr 1, 2021. It is now read-only.

Use tensorization for avx2 as well. #126

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

kimishpatel
Copy link
Contributor

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

Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:
@kimishpatel
Copy link
Contributor Author

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.

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants