Skip to content

rocSPARSE support#463

Merged
pxl-th merged 51 commits intomasterfrom
pxl-th/rocsparse
Aug 8, 2023
Merged

rocSPARSE support#463
pxl-th merged 51 commits intomasterfrom
pxl-th/rocsparse

Conversation

@pxl-th
Copy link
Member

@pxl-th pxl-th commented Aug 8, 2023

Supersedes #298.
Main differences from #298:

  • Use TLS library state, stream, etc. Fixes issues with synchronization.
  • Remove wait/mark adaptors, they are not used anymore.
  • Remove duplicate KernelState.
  • Fix generic.jl tests (init with zero dense arrays used for storing mv! output).
  • Include rocSPARSE tests in the main test suite.
  • Minor refactor for style consistency.

Note: certain operations, like broadcasting two rocsparse arrays require accumulate! kernel, which AMDGPU does not yet have (it performs scalar indexing instead). Disabled respective tests until we have it implemented.

CC: @matinraayai

jpsamaroo and others added 30 commits August 8, 2023 15:44
(cherry picked from commit 62cf82d)
(cherry picked from commit ed48276)
(cherry picked from commit fbf4f60)
(cherry picked from commit 59894d3)
(cherry picked from commit 1fe5a41)
(cherry picked from commit 95727ae)
(cherry picked from commit 1a5b421)
(cherry picked from commit 6ee90d3)
(cherry picked from commit 16f977b)
(cherry picked from commit ef683ab)
(cherry picked from commit b416bc4)
(cherry picked from commit b5b7ade)
(cherry picked from commit c7cc616)
(cherry picked from commit 41d2d41)
(cherry picked from commit 68d0c37)
(cherry picked from commit e26be82)
(cherry picked from commit ecd9e8a)
(cherry picked from commit b62e265)
(cherry picked from commit 75b94bc)
(cherry picked from commit ad23956)
(cherry picked from commit da60fc6)
(cherry picked from commit 9237b9c)
(cherry picked from commit c7d7b40)
(cherry picked from commit cfba911)
@pxl-th pxl-th force-pushed the pxl-th/rocsparse branch from 5a0c0d0 to cd817a1 Compare August 8, 2023 12:47
@pxl-th
Copy link
Member Author

pxl-th commented Aug 8, 2023

Let's merge it to have some initial support, and implement accumulate! in a separate PR.

@pxl-th pxl-th merged commit a557574 into master Aug 8, 2023
@pxl-th pxl-th deleted the pxl-th/rocsparse branch August 8, 2023 13:24
@matinraayai
Copy link
Contributor

matinraayai commented Aug 8, 2023

@pxl-th @jpsamaroo I'd never thought I'd see the day :D. Thanks for working hard to merge this.

@pxl-th
Copy link
Member Author

pxl-th commented Aug 8, 2023

@pxl-th @jpsamaroo I'd never thought I'd see the day :D. Thanks for working hard to merge this.

Thank you for putting a PR in the first place!

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.

3 participants