赞
踩
ICAMAX finds the index of the first element having maximum |Re(.)| + |Im(.)|
也就是找出一列 complex 类型的数据中,实部的绝对值 + 虚部的绝对值 最大的那个元素。这样选出来的主元也还不错。
而不是选择复数模最大的元素;
但是,cublas中的cublasCge-t-rf-Batched()这个函数的kernel,选主元时,使用的是模的最大值;
float A_data[2*4*4]={
-1.640400, 0.086700, 0.114300, 0.869600, -1.195400, 0.486000, -0.824600, 1.210900,
0.696700, 0.978500, -1.112300, 0.214200, -0.090400, 0.757800, -1.382100, 2.060100,
-0.513600, 0.106000, 2.029900, 1.663000, -0.847300, 0.601800, 1.036800, 0.824300,
1.614700, 0.034300, -0.918400, 1.338800, -0.762500, 0.142100, -0.284000, 0.894200
};
cuComplex * A = (cuComplex* )A_data;
cublasCgetrfBatched(A);
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。