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

[SYSTEMDS-3375] CUDA 11.x / CUDNN 8.x support #1610

Draft
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

corepointer
Copy link
Contributor

This changeset brings CUDA11 and CUDNN8 support to SystemDS. All required API changes to successfully compile & run have been applied. A few library functions are still on the ToDo list. They continue to work under a deprecation notice for the time being.
Tested on CUDA 11.6.1 and CUDNN 8.4

@phaniarnab
Copy link
Contributor

Can you please execute the GPUFullReuseTest and GPULineageCacheEvictionTest tests under the Lineage package as well with the updated CUDA version? You need to uncomment the code in TestUtils.java:isGPUAvailable() to trigger lineage-based reuse in GPU. @corepointer

@corepointer
Copy link
Contributor Author

mvn -q -B test -D maven.test.skip=false -D automatedtestbase.outputbuffering=false -Dtest=org.apache.sysds.test.functions.lineage.GPULineageCacheEvictionTest 2>&1 | tee output-GPULineageCacheEvictionTest.txt worked just fine.

This changeset brings CUDA11 and CUDNN8 support to SystemDS. All required API changes to successfully compile & run have been applied. A few library functions are still on the ToDo list. They continue to work under a deprecation notice for the time being.
Tested on CUDA 11.6.1 and CUDNN 8.4
CUDA ops and code gen native support compiled on Ubuntu20/CUDA-11.6.1/CUDNN-8.4.0
@corepointer corepointer marked this pull request as draft May 12, 2022 22:38
@j143 j143 added this to the systemds-3.2.0 milestone Dec 4, 2023
@j143 j143 modified the milestones: systemds-3.2.0, next-release Feb 8, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
Status: Unplanned
3 participants