Skip to content

About

底层张量运算核函数源码,用于Tensorplusplus上层调用

Resources

Stars

0 stars

Watchers

0 watching

Forks

Latest commit

 

History

6 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 

Repository files navigation

cu_plusplus

张量操作底层api,用于Tensorplusplus底层cuda核函数操作,使用c++17标准,cu12.8版本nvcc编译

调用英伟达cublas和cudnn加速批量矩阵乘法和批量2D卷积

dll编译: h调用 -> cudnn.h cublas_v2.h curand.h cuda_bf16.h cuda_fp16.h

lib调用-> cudart_static.lib

kernel32.lib

user32.lib

gdi32.lib

winspool.lib

comdlg32.lib

advapi32.lib

shell32.lib

ole32.lib

oleaut32.lib

uuid.lib

odbc32.lib

odbccp32.lib

curand.lib

cudart.lib

cublas.lib

cudnn.lib

About

底层张量运算核函数源码,用于Tensorplusplus上层调用

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages