cutlass icon indicating copy to clipboard operation
cutlass copied to clipboard

[QST] Are there plans to add specialisations for Sm90?

Open joerowell opened this issue 1 year ago • 7 comments

What is your question? I recently tried to change the type tags on the DGEMM examples to cutlass::arch::Sm90, which caused a load of compile errors. This is primarily because there's certain template specialisations missing for Sm90, e.g there's no specialisation like this.

Is this something that's on the plans to be added / are PRs welcome?

joerowell avatar Oct 04 '23 14:10 joerowell

the link points to dual gemm example not dgemm. do you want dual gemm or dgemm?

hwu36 avatar Oct 04 '23 14:10 hwu36

Ah sorry, laxness on my part. I meant dual GEMM.

On Wed, 4 Oct 2023, 15:58 Haicheng Wu, @.***> wrote:

the link points to dual gemm example not dgemm. do you want dual gemm or dgemm?

— Reply to this email directly, view it on GitHub https://github.com/NVIDIA/cutlass/issues/1123#issuecomment-1747028826, or unsubscribe https://github.com/notifications/unsubscribe-auth/AF7CASFCIQHJBUBAE42GUETX5V2SPAVCNFSM6AAAAAA5SXO35GVHI2DSMVQWIX3LMV43OSLTON2WKQ3PNVWWK3TUHMYTONBXGAZDQOBSGY . You are receiving this because you authored the thread.Message ID: @.***>

joerowell avatar Oct 04 '23 15:10 joerowell

@danthe3rd

you should be able to still run this example on h100 without changing the source code. just use sm90 in nvcc/cmake command line.

this example does not use new features of hopper. we are looking to write a hopper dedicated version using cutlass 3.x

hwu36 avatar Oct 04 '23 16:10 hwu36

Note: we with @joerowell and others are working on a diverged hard fork of xformers that has little to do with the mainstream :wink: We are rewriting the orignal kernels, doing a lot of optimizations, fixing bugs, etc.

vmarkovtsev avatar Oct 05 '23 08:10 vmarkovtsev

@joerowell optimizations for dual GEMM are on our roadmap for H1'24.

mnicely avatar Oct 10 '23 14:10 mnicely

This issue has been labeled inactive-30d due to no recent activity in the past 30 days. Please close this issue if no further response or action is needed. Otherwise, please respond with a comment indicating any updates or changes to the original issue and/or confirm this issue still needs to be addressed. This issue will be labeled inactive-90d if there is no activity in the next 60 days.

github-actions[bot] avatar Nov 18 '23 20:11 github-actions[bot]

This issue has been labeled inactive-90d due to no recent activity in the past 90 days. Please close this issue if no further response or action is needed. Otherwise, please respond with a comment indicating any updates or changes to the original issue and/or confirm this issue still needs to be addressed.

github-actions[bot] avatar Feb 16 '24 20:02 github-actions[bot]