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

Need to select TPL algorithms #2445

Open
jczhang07 opened this issue Dec 2, 2024 · 1 comment
Open

Need to select TPL algorithms #2445

jczhang07 opened this issue Dec 2, 2024 · 1 comment

Comments

@jczhang07
Copy link
Contributor

More specifically, if TPL CUDA is enabled, I want to try various cusparse_spgemm algorithms, including

    CUSPARSE_SPGEMM_DEFAULT                 = 0,
    CUSPARSE_SPGEMM_CSR_ALG_DETERMINITIC    = 1,
    CUSPARSE_SPGEMM_CSR_ALG_NONDETERMINITIC = 2,
    CUSPARSE_SPGEMM_ALG1                    = 3,
    CUSPARSE_SPGEMM_ALG2                    = 4,
    CUSPARSE_SPGEMM_ALG3                    = 5

but KK's SPGEMMAlgorithm does not support them (except DEFAULT).

@lucbv lucbv self-assigned this Jan 20, 2025
@lucbv
Copy link
Contributor

lucbv commented Jan 20, 2025

Starting to look into this. Might want to change the handle and the no_reuse version of the TPL implementation layer.
I don't think these are callable from the spgemm implementation that allows reuse.
It looks like we will need to check for CUDA_VERSION >= 12001.
Finally, we will need a couple of implementation for the regular variants and the mem variants.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

2 participants