cuda_example

cuda_example

CI	状态
pip 编译
轮子
一个使用 [pybind11][], [CUDA][] 和 [numpy][] 构建的示例项目
[scikit-build-core][]。Python 3.9+。
扩展以两种方式渲染 [曼德尔布罗特集合][mandelbrot] — — 一次在
CPU 和 GPU 上 — 这样您可以同时阅读两者,并进行比较
性能。两种实现都是故意以相同的方式编写的:

• src/mandelbrot_cpu.cpp — 一个普通的嵌套循环，遍历每个像素
• src/mandelbrot.cu — 与 CUDA 内核的逻辑相同，每个像素一个线程

CUDA 工具包 (nvcc) 必须已安装且可由 CMake 发现。