📄 atl_tbsvunn.c
字号:
/* --------------------------------------------------------------------- * * -- Automatically Tuned Linear Algebra Software (ATLAS) * (C) Copyright 2000 All Rights Reserved * * -- ATLAS routine -- Version 3.2 -- December 25, 2000 * * Author : Antoine P. Petitet * Contributor(s) : R. Clint Whaley * Originally developed at the University of Tennessee, * Innovative Computing Laboratory, Knoxville TN, 37996-1301, USA. * * --------------------------------------------------------------------- * * -- Copyright notice and Licensing terms: * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions, and the following disclaimer in * the documentation and/or other materials provided with the distri- * bution. * 3. The name of the University, the ATLAS group, or the names of its * contributors may not be used to endorse or promote products deri- * ved from this software without specific written permission. * * -- Disclaimer: * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE UNIVERSITY * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPE- * CIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED * TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, * OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEO- * RY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (IN- * CLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * * --------------------------------------------------------------------- *//* * Include files */#include "atlas_misc.h"#include "atlas_mv.h"#include "atlas_r1.h"#include "atlas_level1.h"#include "atlas_kernel2.h"#include "atlas_reflvl2.h"#include "atlas_lvl2.h"#ifdef TREAL#define MinN_tbsvUNN 16#else#define MinN_tbsvUNN 8#endifvoid Mjoin( PATL, tbsvUNN )( const int N, const int K, const TYPE * A, const int LDA, TYPE * X){/* * Purpose * ======= * * Mjoin( PATL, tbsvUNN ) solves the following triangular system of equations * * A * x = b, * * where b and x are n-element vectors and A is an n by n non-unit, up- * per triangular band matrix, with ( k + 1 ) diagonals. * * No test for singularity or near-singularity is included in this * routine. Such tests must be performed before calling this routine. * * This is a recursive version of the algorithm. For a more detailed * description of the arguments of this function, see the reference im- * plementation in the ATLAS/src/blas/reference directory. * * --------------------------------------------------------------------- *//* * .. Local Variables .. */#ifdef TREAL#define none ATL_rnone#define one ATL_rone#define lda2 LDA#else const TYPE none[2] = { ATL_rnone, ATL_rzero }, one [2] = { ATL_rone, ATL_rzero }; const int lda2 = ( LDA << 1 );#endif int ia, kl, ku, ma, na, n1, n1s, n2;/* .. * .. Executable Statements .. * */ if( N <= MinN_tbsvUNN ) { Mjoin( PATL, reftbsvUNN )( N, K, A, LDA, X, 1 ); } else { n2 = N - ( n1 = ( N >> 1 ) ); n1s = (n1 SHIFT); Mjoin( PATL, tbsvUNN )( n2, K, A+n1*lda2, LDA, X+n1s ); ia = n1 - K; ia = Mmax( ia, 0 ); ma = n1 - ia; na = Mmin( n2, K ); kl = ma - 1; kl = Mmax( kl, 0 ); ku = K - 1 - kl; ku = Mmax( ku, 0 ); Mjoin( PATL, gbmv )( AtlasNoTrans, ma, na, kl, ku, none, A+n1*lda2, LDA, X+n1s, 1, one, X+(ia SHIFT), 1 ); Mjoin( PATL, tbsvUNN )( n1, K, A, LDA, X ); }/* * End of Mjoin( PATL, tbsvUNN ) */}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -