类 cubase

继承关系

派生类型

类文件

class cubase

数据类型的基类

Subclassed by cumat, custride

公共类型

enum class Type

Values:

enumerator Base
enumerator Op
enumerator Mat
enumerator Stride
enumerator Batched
enum class Init

Values:

enumerator None
enumerator Zero

公共职能

inline cubase()

构造一个新的 cubase 对象。

inline cubase(size_t bytes, Init init = Init::Zero)

构造一个新的 cubase 对象。

参数:
  • bytes -- 字节数

  • init -- 初始化方式

inline virtual ~cubase()

销毁 cubase 对象。

virtual size_t nbytes() const = 0

获取字节数

返回:

size_t 字节数

inline double *dmem() const

获取数据指针

返回:

double* 数据指针

inline virtual void get(double *dst)

提取 GPU 中的数据

参数:

dst -- 指向要保存数据位置的指针

Public Static Functions

static inline auto create_handle()
static inline auto destory_handle()

公共静态属性

static cublasHandle_t handle

Save handle for cublas

static constexpr const double alpha1 = 1.0
static constexpr const double beta0 = 0.0
static constexpr const double beta1 = 1.0
static constexpr cuop::Op op = cuop::Op::Origin
static constexpr cubase::Type type = cubase::Type::Base

受保护属性

bool mIsRelease = false
double *dMem = nullptr