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

hidet-org / hidet

An open-source efficient deep learning framework/compiler, written in python.

https://hidet.org

Apache License 2.0

634 stars 50 forks source link

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

Closed yudi0201 closed 5 months ago

yaoyaoding commented 5 months ago

Thanks @yudi0201 !