search

NVIDIA / cuda-samples

Samples for CUDA Developers which demonstrates features in CUDA Toolkit
Other
6.1k stars 1.77k forks source link

cudaGraph and cublas #62

Open zkh2016 opened 3 years ago

zkh2016 commented 3 years ago

how to add cublas sgemm to cudaGraph by cudaGraphAddKernelNode?

pmixer commented 2 years ago

how to add cublas sgemm to cudaGraph by cudaGraphAddKernelNode?

https://docs.nvidia.com/cuda/cuda-samples/index.html

Added 7_CUDALibraries/conjugateGradientCudaGraphs. Demonstrates conjugate gradient solver on GPU using CUBLAS/CUSPARSE library calls captured and called using CUDA Graph APIs.

https://github.com/NVIDIA/cuda-samples/blob/master/Samples/conjugateGradientCudaGraphs/conjugateGradientCudaGraphs.cu