有没有办法为 CuPy 计算设置线程数、块数和网格数?如何?
Is there any way to set number of threads , number of blocks and grids for CuPy computation? How?
我正在使用带有以下代码的 Cupy,
import cupy as cp
vals, vecs = cp.linalg.eigh(Array)
对于上面的代码,我想在代码中设置线程数。怎么做。让我们假设 100 个线程执行代码。
对于类似于 NumPy 的高级 API,目前没有 public 接口来更改 grid/block 配置。此外,许多 linalg API(例如您示例中的 eigh
)将作业委托给 CUDA 数学库求解器,它们也不允许用户设置 grid/block 配置。我想知道是什么提示了这种需求。如果能详细点就好了
我正在使用带有以下代码的 Cupy,
import cupy as cp
vals, vecs = cp.linalg.eigh(Array)
对于上面的代码,我想在代码中设置线程数。怎么做。让我们假设 100 个线程执行代码。
对于类似于 NumPy 的高级 API,目前没有 public 接口来更改 grid/block 配置。此外,许多 linalg API(例如您示例中的 eigh
)将作业委托给 CUDA 数学库求解器,它们也不允许用户设置 grid/block 配置。我想知道是什么提示了这种需求。如果能详细点就好了