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

Add cuSPARSELt to the nightlies #110168

Closed
jcaip opened this issue Sep 27, 2023 · 4 comments
Closed

Add cuSPARSELt to the nightlies #110168

jcaip opened this issue Sep 27, 2023 · 4 comments
Assignees
Labels
module: binaries Anything related to official binaries that we release to users module: build Build system issues triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module

Comments

@jcaip
Copy link
Contributor

jcaip commented Sep 27, 2023

Issue description

cuSPASRELt is a library that offers accelerated semi-structured sparse matmul. We are looking to integrate v0.5.0
The download link can be found here: https://developer.nvidia.com/cusparselt-downloads
The library is ~ 20 Mb.

We merged build support for cuSPASRELt in this PR
To build pytorch with cuSPASRELt, we need to set two env variables,

USE_CUSPARSELT=1
CUSPARSELT_ROOT=/path/to/cusparselt/download

cc @seemethere @malfet

@malfet malfet added module: binaries Anything related to official binaries that we release to users module: build Build system issues triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module labels Sep 27, 2023
@jcaip
Copy link
Contributor Author

jcaip commented Oct 4, 2023

Hi just FYI the cusparselt version has been bumped to v0.5.0, we'd like to use the newest version

@malfet
Copy link
Contributor

malfet commented Oct 11, 2023

Updates:

/opt/rh/devtoolset-9/root/usr/libexec/gcc/x86_64-redhat-linux/9/ld: /usr/local/cuda/lib64/libcusparseLt.so.0: undefined reference to `std::__cxx11::basic_string<char, std::char_traits<char>, std::allocator<char> >::_M_construct(unsigned long, char)@GLIBCXX_3.4.21'

@malfet
Copy link
Contributor

malfet commented Oct 11, 2023

Ok, but cuSparseLT-0.4.0 seems to be compatible with CentOS7, let's try that...

@malfet
Copy link
Contributor

malfet commented Oct 31, 2023

Added to nightly wheels, not sure if/when we need it for conda, but closing for now

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
module: binaries Anything related to official binaries that we release to users module: build Build system issues triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module
Projects
Archived in project
Development

No branches or pull requests

2 participants