cuBLAS中的cublasSetVector()和cublasGetVector()

2.4.9. cublasGetStream()

cublasStatus_t
cublasGetStream(cublasHandle_t handle, cudaStream_t *streamId)
1
2

此函数获取 cuBLAS 库流，该流用于执行对 cuBLAS 库函数的所有调用。 如果未设置 cuBLAS 库流，则所有内核都使用 default NULL 流。

2.4.10. cublasGetPointerMode()

cublasStatus_t
cublasGetPointerMode(cublasHandle_t handle, cublasPointerMode_t *mode)
1
2

该函数获取cuBLAS库使用的指针模式。 有关更多详细信息，请参阅有关 cublasPointerMode_t 类型的部分。

2.4.11. cublasSetPointerMode()

cublasStatus_t
cublasSetPointerMode(cublasHandle_t handle, cublasPointerMode_t mode)
1
2

此函数设置 cuBLAS 库使用的指针模式。 默认值是通过主机上的引用传递的值。 有关更多详细信息，请参阅有关 cublasPointerMode_t 类型的部分。

2.4.12. cublasSetVector()

cublasStatus_t
cublasSetVector(int n, int elemSize,
                const void *x, int incx, void *y, int incy)
1
2
3

此函数将 n 个元素从主机内存空间中的向量 x 复制到 GPU 内存空间中的向量 y。 假定两个向量中的元素的大小为 elemSize 字节。 连续元素之间的存储间距由源向量 x 的 incx 和目标向量 y 的 incy 给出。

由于假定二维矩阵的列主要格式，如果向量是矩阵的一部分，则等于 1 的向量增量访问该矩阵的（部分）列。 类似地，使用等于矩阵前导维度的增量会导致访问该矩阵的（部分）行。

2.4.13. cublasGetVector()

cublasStatus_t
cublasGetVector(int n, int elemSize,
                const void *x, int incx, void *y, int incy)
1
2
3

此函数将 n 个元素从 GPU 内存空间中的向量 x 复制到主机内存空间中的向量 y。 假定两个向量中的元素的大小为 elemSize 字节。 连续元素之间的存储间距由源向量的 incx 和目标向量 y 的 incy 给出。

由于假定二维矩阵的列主要格式，如果向量是矩阵的一部分，则等于 1 的向量增量访问该矩阵的（部分）列。 类似地，使用等于矩阵前导维度的增量会导致访问该矩阵的（部分）行。