pycuda.driver.mem_alloc()在哪里？在pycuda/driver.py中，我只能找到mem_alloc_like()
pycuda.driver.mem_alloc()在哪里？在pycuda/driver.py中，我只能找到mem_alloc_like()
提问于 2018-08-23 07:05:31
我对python和PYCUDA很陌生。
在许多PYCUDA的例子中，我可以找到这样的代码：
import pycuda.driver as cuda
cuda.mem_alloc(a.nbytes) 但是，当我查看pycuda/driver.py时，我无法找到定义mem_alloc的位置。
我只能找到mem_alloc_like，它调用mem_alloc()：
def mem_alloc_like(ary):
return mem_alloc(ary.nbytes)这告诉我，mem_alloc必须在库中的某个地方，driver.py导入。
但是，下面是我在driver.py中找到的导入，6个、numpy或sys中没有一个函数名为"mem_alloc()“
from __future__ import absolute_import
from __future__ import print_function
import six
try:
from pycuda._driver import * # noqa
except ImportError as e:
if "_v2" in str(e):
from warnings import warn
warn("Failed to import the CUDA driver interface, with an error "
"message indicating that the version of your CUDA header "
"does not match the version of your CUDA driver.")
raise
import numpy as np
import sys回答 1
Stack Overflow用户
回答已采纳
发布于 2018-08-23 09:24:19
pyCUDA是CUDA驱动程序API的包装器。这就是pyCUDA公开这些APIs的方式。
- 用C/ C++编写驱动程序API的包装器
- 使用Boost::python向python公开这些函数，因为python的解释器是用C(Cpython)实现的
- 将这些包装器编译成一个共享库，即
_driver.so。 - 导入共享库，即
from pycuda._driver import *。
但是，当我查看pycuda/driver.py时，我无法找到定义mem_alloc的位置。
它在pycuda/src/cpp/cuda.hpp中定义。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/51980087
复制相关文章
相似问题
腾讯云开发者
