当前位置:   article > 正文

部分选主元 LU 分解,CGETRF in lapack 选主元时调用 ICAMAX_lu分解 cublascgetrfbatched

lu分解 cublascgetrfbatched

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);

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/知新_RL/article/detail/245541
推荐阅读
相关标签
  

闽ICP备14008679号