ocean::CompileOption
大约 1 分钟
ocean::CompileOption
编译选项 More...
#include <Option.hpp>
Public Functions
Name | |
---|---|
CompileOption() 默认选项,仅CPU后端 | |
CompileOption(DeviceBackend backend) 特定后端的编译选项 | |
void | setCpuThreadNumber(int number) 设置CPU线程数量 |
int | getCpuThreadNumber() 获取CPU线程数量 |
void | setCudaDevices(const std::string & devices) 设置CUDA设备 |
std::set< int > | getCudaDevices() 获取CUDA设备列表 |
CompileOption | cpu() CPU 默认编译选项 |
CompileOption | cuda() CUDA 默认编译选项 |
Public Attributes
Name | |
---|---|
Arguments | arguments 参数载体 |
std::vector< DeviceBackend > | backendStack 后端设备执行堆栈 |
Detailed Description
struct ocean::CompileOption;
编译选项
编译选择使用 Arguments 作为主要的存储载体,具有良好的二进制兼容性及扩展性
Public Functions Documentation
function CompileOption
CompileOption()
默认选项,仅CPU后端
function CompileOption
explicit CompileOption(
DeviceBackend backend
)
特定后端的编译选项
Parameters:
- backend
function setCpuThreadNumber
void setCpuThreadNumber(
int number
)
设置CPU线程数量
Parameters:
- number 线程数量
function getCpuThreadNumber
int getCpuThreadNumber()
获取CPU线程数量
Return: 线程数量
function setCudaDevices
void setCudaDevices(
const std::string & devices
)
设置CUDA设备
Parameters:
- devices 例如:
1. "0, 1, 2, 3"
2. "0"
3. "0,"
4. "all"
function getCudaDevices
std::set< int > getCudaDevices()
获取CUDA设备列表
Return: CUDA设备集合
function cpu
static CompileOption cpu()
CPU 默认编译选项
Return: 编译选项
function cuda
static CompileOption cuda()
CUDA 默认编译选项
Return: 编译选项
Public Attributes Documentation
variable arguments
Arguments arguments;
参数载体
variable backendStack
std::vector< DeviceBackend > backendStack;
后端设备执行堆栈
Updated on 2022-08-19 at 17:14:26 +0800