gpupy:cuBlas\NumbaPro 函数的包装器,提供对 GPU 上 NumPy 函数的支持

时间:2024-07-19 10:48:35
【文件属性】:

文件名称:gpupy:cuBlas\NumbaPro 函数的包装器,提供对 GPU 上 NumPy 函数的支持

文件大小:17KB

文件格式:ZIP

更新时间:2024-07-19 10:48:35

Python

小狗 用于 GPU cuBLAS\NumbaPro 函数的包装器,提供对 NumPy 函数的支持 要求: NumPy NumbaPro CUDA安装 支持 CUDA 的 GPU 鼻子 用 实例化一个 Gpupy 对象 from gpupy import Gpupy gp = Gpupy () out = gp . function ( * args ) 职能: gp.dot: np.dot drop-in(矩阵-矩阵和矩阵-向量产品) gp.T:np.transpose 插入(矩阵) gp.add: np.add drop-in(矩阵-矩阵和向量-向量加法) gp.mult: np.multiply drop-in(矩阵-矩阵和向量-向量乘法) Tesla K40 与 Xeon E-5 1620(使用 MKL)的加速比 20x 点 30x 用于添加 Parameters


【文件预览】:
gpupy-master
----timing.py(4KB)
----tests()
--------test_mult.py(677B)
--------test_check.py(377B)
--------test_tanh.py(1KB)
--------test_reshape.py(665B)
--------test_relu.py(4KB)
--------test_sub.py(1KB)
--------test_zero_diag.py(383B)
--------test_ones.py(640B)
--------test_diag.py(483B)
--------test_dot.py(2KB)
--------test_scal.py(648B)
--------test_sum.py(972B)
--------test_zeros.py(998B)
--------test_clip.py(1KB)
--------test_add.py(7KB)
--------test_T.py(310B)
--------test_abs.py(1KB)
--------test_mean.py(995B)
--------test_exp.py(1KB)
----LICENSE(1KB)
----__init__.py(45KB)
----.gitignore(544B)
----README.md(1KB)

网友评论