systemds icon indicating copy to clipboard operation
systemds copied to clipboard

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

Open corepointer opened this issue 3 years ago • 2 comments

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

corepointer avatar May 10 '22 16:05 corepointer

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

phaniarnab avatar May 10 '22 18:05 phaniarnab

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.

corepointer avatar May 10 '22 23:05 corepointer