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.

Sign up for GitHub

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

Jump to bottom

[cuBLAS] Add cublas_gemm_batched and use cublasSetStream to set stream to the current stream in all cublas API calls #423

Merged
merged 4 commits into from
Feb 12, 2024
Merged

[cuBLAS] Add cublas_gemm_batched and use cublasSetStream to set stream to the current stream in all cublas API calls #423

merged 4 commits into from
Feb 12, 2024

Commits on Jan 31, 2024

  1. [cublas] Add cublas_gemm_batched

    Yudi Sun committed Jan 31, 2024
    Configuration menu
    Copy the full SHA
    3d58fbd View commit details
    Browse the repository at this point in the history

Commits on Feb 6, 2024

  1. Use cudaMallocAsync and cudaMemcpyAsync for cublas_batched_gemm

    Yudi Sun committed Feb 6, 2024
    Configuration menu
    Copy the full SHA
    a446270 View commit details
    Browse the repository at this point in the history

Commits on Feb 7, 2024

  1. Use cublasSetStream to set the stream to the current stream in all cu… 

    …blas API calls
    Yudi Sun committed Feb 7, 2024
    Configuration menu
    Copy the full SHA
    e4ec611 View commit details
    Browse the repository at this point in the history

Commits on Feb 10, 2024

  1. [cublas] Use cuda_free_async

    Yudi Sun committed Feb 10, 2024
    Configuration menu
    Copy the full SHA
    87bf01d View commit details
    Browse the repository at this point in the history