【文件属性】:
文件名称:gpupy:cuBlas\NumbaPro 函数的包装器,提供对 GPU 上 NumPy 函数的支持
文件大小:17KB
文件格式:ZIP
更新时间:2021-06-29 17:01:55
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)