标签：cuda

  • 用于CUDA的Numba.vectorize:返回数组的正确签名是什么？ - python

    时间：2020-8-15

    我具有以下结构的功能， @numba.jit(nopython = True) def foo(X,N): ''' :param X: 1D numpy array :param N: Integer :rtype: 2D numpy array of shape len(X) x N ''' ou […]

    标签：, ,

  • Python中的自动CudaMat转换 - python

    时间：2020-7-26

    我正在考虑使用某种形式的CUDA来加快我的python代码（全部为矩阵数学）的速度。目前，我的代码使用的是Python和Numpy，因此使用PyCUDA或CudaMat之类的代码重写它似乎并不难。 但是，在我第一次尝试使用CudaMat时，我意识到我必须重新排列许多方程式才能将所有操作保留在GPU上。这包括创建许多临时变量，以便我可以存储操作结果。 我理解为 […]

    标签：, , , ,