有没有办法为 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 配置。我想知道是什么提示了这种需求。如果能详细点就好了