ocean::CompileOption

Neurocean大约 1 分钟

ocean::CompileOption

编译选项 More...

#include <Option.hpp>

Public Functions

Name
CompileOption()
默认选项,仅CPU后端
CompileOption(DeviceBackend backend)
特定后端的编译选项
voidsetCpuThreadNumber(int number)
设置CPU线程数量
intgetCpuThreadNumber()
获取CPU线程数量
voidsetCudaDevices(const std::string & devices)
设置CUDA设备
std::set< int >getCudaDevices()
获取CUDA设备列表
CompileOptioncpu()
CPU 默认编译选项
CompileOptioncuda()
CUDA 默认编译选项

Public Attributes

Name
Argumentsarguments
参数载体
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

上次编辑于:
贡献者: damone