结构体 IParallelCudaEnabled

结构体文档

struct IParallelCudaEnabled

公有成员函数

virtual void setGPUId(const int gpuId) = 0

设置并行化使用的 GPU 的 ID。

参数

gpuId – 并行化使用的 GPU 的 ID。 从 0 开始。

virtual void setGroupSize(const double size) = 0

Set the group size while paralleling.

参数

size – 并行化时的组长度 \(g\)。 通常取决于 gpuID 对应显卡的显存 \(m\) (B)。 如果有 \(n\) 样本和 \(k\) 变量, 要求

\[ k * n * g * 8 < m \]
。 对于大多数 GPU 可选择值 64。