结构体 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。
-
virtual void setGPUId(const int gpuId) = 0