cublasIsamin的用法

此处省略链接库操作

目录

前言

一、Cublas官网

二、使用步骤

1.函数解释

2.用法

总结
 

---

前言

Q1：寻找一个CUDA数组中的最小最大值的索引

Q2：如何调用

一、Cublas官网

详细内容参考
。

二、使用步骤

1.函数解释

代码如下（示例）：打开cublas.h

/*------------------------------------------------------------------------*/
/* AMIN */
int CUBLASWINAPI cublasIsamin (int n, const float *x, int incx);
int CUBLASWINAPI cublasIdamin (int n, const double *x, int incx);

 n：数组的大小   x：CUDA内存中的数组   incx：寻找最小值索引的增量（不是最小值）

2.用法

代码如下（示例）：

int inc = 1;  //索引增量
int minvloc = 0;
cublasInit();
minvloc = cublasIsamin(datasize, mydata_d, inc) - 1;  //索引从1开始，转换到数组-1
cublasShutdown();
HDLOGINFO("minvloc=%d", minvloc);

mydata_d 为提前申请好内存的数组，minvloc为最小值的索引，注意在cublas中的索引从1开始，因此转换回数组中的索引需要减1

---

总结

cublasInit();
调用cublas函数
cublasShutdown();

这个结构为固定结构，使用其他函数如法炮制