Skip to content

[WIP] GEMM: add support for mixed input scalar types#1557

Draft
mzuzek wants to merge 26 commits intokokkos:developfrom
NexGenAnalytics:TODO/gemm-mixed-scalars
Draft

[WIP] GEMM: add support for mixed input scalar types#1557
mzuzek wants to merge 26 commits intokokkos:developfrom
NexGenAnalytics:TODO/gemm-mixed-scalars

Conversation

@mzuzek
Copy link
Copy Markdown

@mzuzek mzuzek commented Sep 29, 2022

WIP: based on #1556 - please review and merge before reviewing this PR


This PR adds support for mixed input scalar types into:

  • GEMM implementation and unit tests:
    • SerialGemm
    • TeamGemm, TeamVectorGemm and ThreadVectorGemm
  • Inner methods:
    • InnerGemmFixA
    • InnerGemmFixB + compiler fixes
    • InnerGemmFixC

Mikołaj Zuzek added 26 commits September 28, 2022 11:16
@mzuzek mzuzek closed this Sep 29, 2022
@mzuzek mzuzek deleted the TODO/gemm-mixed-scalars branch September 29, 2022 21:19
@mzuzek mzuzek restored the TODO/gemm-mixed-scalars branch September 29, 2022 21:19
@mzuzek mzuzek reopened this Sep 29, 2022
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.

1 participant