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

[ROCM] Support rocblas_sgemm_strided_batched #6579

Merged
merged 1 commit into from Sep 29, 2020
Merged

[ROCM] Support rocblas_sgemm_strided_batched #6579

merged 1 commit into from Sep 29, 2020

Conversation

csullivan
Copy link
Contributor

Support rocBLAS batched GEMM and prioritize rocBLAS strategy over codegen when no costs are provided via a tuning config. Also canonicalize invocation of column major rocBLAS APIs for matmul and batched_matmul.

@csullivan
Copy link
Contributor Author

cc @masahi

@masahi masahi self-assigned this Sep 28, 2020
@masahi
Copy link
Member

masahi commented Sep 28, 2020

please take a look at the CI problem

@masahi masahi merged commit 22ad0dd into apache:master Sep 29, 2020
@masahi
Copy link
Member

masahi commented Sep 29, 2020

thanks @csullivan

@csullivan
Copy link
Contributor Author

Thank you @masahi !

TusharKanekiDey pushed a commit to TusharKanekiDey/tvm that referenced this pull request Oct 13, 2020
TusharKanekiDey pushed a commit to TusharKanekiDey/tvm that referenced this pull request Oct 14, 2020
TusharKanekiDey pushed a commit to TusharKanekiDey/tvm that referenced this pull request Oct 15, 2020
TusharKanekiDey pushed a commit to TusharKanekiDey/tvm that referenced this pull request Oct 16, 2020
trevor-m pushed a commit to neo-ai/tvm that referenced this pull request Oct 19, 2020
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.

None yet

2 participants