cuquantum.custatevec.compute_expectation_get_workspace_size¶
- cuquantum.custatevec.compute_expectation_get_workspace_size(intptr_t handle, int sv_data_type, uint32_t n_index_bits, intptr_t matrix, int matrix_data_type, int layout, uint32_t n_basis_bits, int compute_type) size_t[源代码]¶
此函数获取
compute_expectation()所需的工作区大小。- 参数
handle (intptr_t) – cuStateVec 上下文的句柄。
sv_data_type (int) – 状态向量的数据类型。
n_index_bits (uint32_t) – 状态向量的索引位数。
matrix (intptr_t) – 指向矩阵的主机或设备指针。
matrix_data_type (int) – 矩阵的数据类型。
layout (MatrixLayout) – 枚举器，指定矩阵的内存布局。
n_basis_bits (uint32_t) – 目标位的数量。
compute_type (ComputeType) – 矩阵乘法的 compute_type。
- 返回
额外工作区的大小。
- 返回类型
size_t