Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[MKL] Fix offloading of batch_matmul to MKL #6752

Merged
merged 3 commits into from
Oct 25, 2020

Conversation

masahi
Copy link
Member

@masahi masahi commented Oct 25, 2020

In #6182, a new extern lib mkl was added to separate MKL offload from cblas. But x86 batch_matmul schedule was not updated to reflect that change, so adding -libs=mkl won't call mkl batched matmul. (I think before #6182, -libs=cblas will call mkl batch matmul if mkl is enabled)

please review @tkonolige @icemelon9 @anijain2305 @tqchen

Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. Thanks for catching this. Just a nit.

python/tvm/topi/x86/batch_matmul.py Show resolved Hide resolved
@masahi masahi merged commit 2d8ac1d into apache:main Oct 25, 2020
@masahi
Copy link
Member Author

masahi commented Oct 25, 2020

Thanks @comaniac

zhiics pushed a commit to zhiics/tvm that referenced this pull request Oct 28, 2020
* fix mkl offloading of batch matmul

* name fix and add doc

* add doc for lib arg

Co-authored-by: masa <[email protected]>
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Oct 29, 2020
* fix mkl offloading of batch matmul

* name fix and add doc

* add doc for lib arg

Co-authored-by: masa <[email protected]>
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 2, 2020
* fix mkl offloading of batch matmul

* name fix and add doc

* add doc for lib arg

Co-authored-by: masa <[email protected]>
trevor-m pushed a commit to trevor-m/tvm that referenced this pull request Dec 4, 2020
* fix mkl offloading of batch matmul

* name fix and add doc

* add doc for lib arg

Co-authored-by: masa <[email protected]>
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Dec 4, 2020
* fix mkl offloading of batch matmul

* name fix and add doc

* add doc for lib arg

Co-authored-by: masa <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants