📄 atl_hemvu.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"void Mjoin( PATL, hemvU )( const int N, const TYPE * A, const int LDA, const TYPE * X, const SCALAR BETA, TYPE * Y){/* * Purpose * ======= * * Mjoin( PATL, hemvU ) performs the matrix-vector operation * * y := alpha * A * x + beta * y, * * where alpha and beta are scalars, x and y are n-element vectors and A * is an n by n Hermitian matrix. * * 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 .. */ void (*gemvT)( const int, const int, const SCALAR, const TYPE *, const int, const TYPE *, const int, const SCALAR, TYPE *, const int ); void (*gemvN)( const int, const int, const SCALAR, const TYPE *, const int, const TYPE *, const int, const SCALAR, TYPE *, const int );#ifdef TREAL#define one ATL_rone TYPE beta0;#else const TYPE * beta0, one[2] = { ATL_rone, ATL_rzero };#endif TYPE * A0, * x0, * y0; int j, jb, jbs, m, mb, nb;/* .. * .. Executable Statements .. * */ ATL_GetPartSYMV( A, LDA, &mb, &nb ); beta0 = BETA; if( SCALAR_IS_ZERO( beta0 ) ) { gemvN = Mjoin( PATL, gemvS_a1_x1_b0_y1 ); gemvT = Mjoin( PATL, gemvC_a1_x1_b0_y1 ); } else if( SCALAR_IS_ONE ( beta0 ) ) { gemvN = Mjoin( PATL, gemvS_a1_x1_b1_y1 ); gemvT = Mjoin( PATL, gemvC_a1_x1_b1_y1 ); } else { gemvN = Mjoin( PATL, gemvS_a1_x1_bX_y1 ); gemvT = Mjoin( PATL, gemvC_a1_x1_bX_y1 ); } MUrnext( N, A, LDA ); x0 = (TYPE *)(X); X += (N SHIFT); y0 = (TYPE *)(Y); Y += (N SHIFT); for( j = 0; j < N; j += nb ) { jb = N - j; jb = Mmin( jb, nb ); jbs = (jb SHIFT); MUrprev( jb, A, LDA ); X -= jbs; Y -= jbs; if( ( m = N-j-jb ) != 0 ) { A0 = (TYPE *)(A) - (m SHIFT); gemvT( jb, m, one, A0, LDA, x0, 1, beta0, Y, 1 ); gemvN( m, jb, one, A0, LDA, X, 1, beta0, y0, 1 ); beta0 = one; } Mjoin( PATL, refhemvU )( jb, one, A, LDA, X, 1, beta0, Y, 1 ); gemvN = Mjoin( PATL, gemvS_a1_x1_b1_y1 ); gemvT = Mjoin( PATL, gemvC_a1_x1_b1_y1 ); }/* * End of Mjoin( PATL, hemvU ) */}
⌨️ 快捷键说明
复制代码
Ctrl + C
搜索代码
Ctrl + F
全屏模式
F11
切换主题
Ctrl + Shift + D
显示快捷键
?
增大字号
Ctrl + =
减小字号
Ctrl + -