gpt4 book ai didi

python - pyopencl array sum 添加数组

转载 作者:行者123 更新时间:2023-12-01 02:28:28 26 4
gpt4 key购买 nike

我是 opencl 和 pyopencl 的新手。我正在尝试编写一个基本程序来添加。我遇到了this文档并在 python 中尝试了这个小代码。显然,它不起作用。

import pyopencl as cl
import pyopencl.tools
import pyopencl.array
import numpy

context = cl.create_some_context()
queue = cl.CommandQueue(context)

h_a = numpy.random.rand(3,3)
d_a = cl.Buffer(context, cl.mem_flags.READ_ONLY |
cl.mem_flags.COPY_HOST_PTR, hostbuf=h_a)
print cl.array.sum(d_a, dtype=None, queue=queue)

正如您所评估的,我不确定如何使用这些预定义的函数。

最佳答案

我的 PyOpenCl 教程有一个数组求和示例,其中内联注释解释了每行的作用: https://github.com/benshope/PyOpenCL-Tutorial

尝试运行它,希望它有帮助!

关于python - pyopencl array sum 添加数组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47099570/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com