xref: /aosp_15_r20/external/cblas/src/cblas_ztrmv.c (revision 1858f9982ea1ad57fb52080c08594e4d1cce4fa1)
1*1858f998SYi Kong /*
2*1858f998SYi Kong  * cblas_ztrmv.c
3*1858f998SYi Kong  * The program is a C interface to ztrmv.
4*1858f998SYi Kong  *
5*1858f998SYi Kong  * Keita Teranishi  5/20/98
6*1858f998SYi Kong  *
7*1858f998SYi Kong  */
8*1858f998SYi Kong #include "cblas.h"
9*1858f998SYi Kong #include "cblas_f77.h"
cblas_ztrmv(const enum CBLAS_ORDER order,const enum CBLAS_UPLO Uplo,const enum CBLAS_TRANSPOSE TransA,const enum CBLAS_DIAG Diag,const int N,const void * A,const int lda,void * X,const int incX)10*1858f998SYi Kong void cblas_ztrmv(const enum CBLAS_ORDER order, const enum CBLAS_UPLO Uplo,
11*1858f998SYi Kong                  const enum CBLAS_TRANSPOSE TransA, const enum CBLAS_DIAG Diag,
12*1858f998SYi Kong                  const int N, const void  *A, const int lda,
13*1858f998SYi Kong                  void  *X, const int incX)
14*1858f998SYi Kong 
15*1858f998SYi Kong {
16*1858f998SYi Kong    char TA;
17*1858f998SYi Kong    char UL;
18*1858f998SYi Kong    char DI;
19*1858f998SYi Kong #ifdef F77_CHAR
20*1858f998SYi Kong    F77_CHAR F77_TA, F77_UL, F77_DI;
21*1858f998SYi Kong #else
22*1858f998SYi Kong    #define F77_TA &TA
23*1858f998SYi Kong    #define F77_UL &UL
24*1858f998SYi Kong    #define F77_DI &DI
25*1858f998SYi Kong #endif
26*1858f998SYi Kong #ifdef F77_INT
27*1858f998SYi Kong    F77_INT F77_N=N, F77_lda=lda, F77_incX=incX;
28*1858f998SYi Kong #else
29*1858f998SYi Kong    #define F77_N N
30*1858f998SYi Kong    #define F77_lda lda
31*1858f998SYi Kong    #define F77_incX incX
32*1858f998SYi Kong #endif
33*1858f998SYi Kong    int n, i=0, tincX;
34*1858f998SYi Kong    double *st=0,*x=(double *)X;
35*1858f998SYi Kong    extern int CBLAS_CallFromC;
36*1858f998SYi Kong    extern int RowMajorStrg;
37*1858f998SYi Kong    RowMajorStrg = 0;
38*1858f998SYi Kong 
39*1858f998SYi Kong    CBLAS_CallFromC = 1;
40*1858f998SYi Kong    if (order == CblasColMajor)
41*1858f998SYi Kong    {
42*1858f998SYi Kong       if (Uplo == CblasUpper) UL = 'U';
43*1858f998SYi Kong       else if (Uplo == CblasLower) UL = 'L';
44*1858f998SYi Kong       else
45*1858f998SYi Kong       {
46*1858f998SYi Kong          cblas_xerbla(2, "cblas_ztrmv","Illegal Uplo setting, %d\n", Uplo);
47*1858f998SYi Kong          CBLAS_CallFromC = 0;
48*1858f998SYi Kong          RowMajorStrg = 0;
49*1858f998SYi Kong          return;
50*1858f998SYi Kong       }
51*1858f998SYi Kong       if (TransA == CblasNoTrans) TA = 'N';
52*1858f998SYi Kong       else if (TransA == CblasTrans) TA = 'T';
53*1858f998SYi Kong       else if (TransA == CblasConjTrans) TA = 'C';
54*1858f998SYi Kong       else
55*1858f998SYi Kong       {
56*1858f998SYi Kong          cblas_xerbla(3, "cblas_ztrmv","Illegal TransA setting, %d\n", TransA);
57*1858f998SYi Kong          CBLAS_CallFromC = 0;
58*1858f998SYi Kong          RowMajorStrg = 0;
59*1858f998SYi Kong          return;
60*1858f998SYi Kong       }
61*1858f998SYi Kong       if (Diag == CblasUnit) DI = 'U';
62*1858f998SYi Kong       else if (Diag == CblasNonUnit) DI = 'N';
63*1858f998SYi Kong       else
64*1858f998SYi Kong       {
65*1858f998SYi Kong          cblas_xerbla(4, "cblas_ztrmv","Illegal Diag setting, %d\n", Diag);
66*1858f998SYi Kong          CBLAS_CallFromC = 0;
67*1858f998SYi Kong          RowMajorStrg = 0;
68*1858f998SYi Kong          return;
69*1858f998SYi Kong       }
70*1858f998SYi Kong       #ifdef F77_CHAR
71*1858f998SYi Kong          F77_UL = C2F_CHAR(&UL);
72*1858f998SYi Kong          F77_TA = C2F_CHAR(&TA);
73*1858f998SYi Kong          F77_DI = C2F_CHAR(&DI);
74*1858f998SYi Kong       #endif
75*1858f998SYi Kong       F77_ztrmv( F77_UL, F77_TA, F77_DI, &F77_N, A, &F77_lda, X,
76*1858f998SYi Kong                       &F77_incX);
77*1858f998SYi Kong    }
78*1858f998SYi Kong    else if (order == CblasRowMajor)
79*1858f998SYi Kong    {
80*1858f998SYi Kong       RowMajorStrg = 1;
81*1858f998SYi Kong       if (Uplo == CblasUpper) UL = 'L';
82*1858f998SYi Kong       else if (Uplo == CblasLower) UL = 'U';
83*1858f998SYi Kong       else
84*1858f998SYi Kong       {
85*1858f998SYi Kong          cblas_xerbla(2, "cblas_ztrmv","Illegal Uplo setting, %d\n", Uplo);
86*1858f998SYi Kong          CBLAS_CallFromC = 0;
87*1858f998SYi Kong          RowMajorStrg = 0;
88*1858f998SYi Kong          return;
89*1858f998SYi Kong       }
90*1858f998SYi Kong 
91*1858f998SYi Kong       if (TransA == CblasNoTrans) TA = 'T';
92*1858f998SYi Kong       else if (TransA == CblasTrans) TA = 'N';
93*1858f998SYi Kong       else if (TransA == CblasConjTrans)
94*1858f998SYi Kong       {
95*1858f998SYi Kong          TA = 'N';
96*1858f998SYi Kong          if ( N > 0)
97*1858f998SYi Kong          {
98*1858f998SYi Kong             if(incX > 0)
99*1858f998SYi Kong                tincX = incX;
100*1858f998SYi Kong             else
101*1858f998SYi Kong                tincX = -incX;
102*1858f998SYi Kong             i = tincX << 1;
103*1858f998SYi Kong             n = i * N;
104*1858f998SYi Kong             x++;
105*1858f998SYi Kong             st = x + n;
106*1858f998SYi Kong             do
107*1858f998SYi Kong             {
108*1858f998SYi Kong                *x = -(*x);
109*1858f998SYi Kong                x += i;
110*1858f998SYi Kong             }
111*1858f998SYi Kong             while (x != st);
112*1858f998SYi Kong             x -= n;
113*1858f998SYi Kong          }
114*1858f998SYi Kong       }
115*1858f998SYi Kong       else
116*1858f998SYi Kong       {
117*1858f998SYi Kong          cblas_xerbla(3, "cblas_ztrmv","Illegal TransA setting, %d\n", TransA);
118*1858f998SYi Kong          CBLAS_CallFromC = 0;
119*1858f998SYi Kong          RowMajorStrg = 0;
120*1858f998SYi Kong          return;
121*1858f998SYi Kong       }
122*1858f998SYi Kong 
123*1858f998SYi Kong       if (Diag == CblasUnit) DI = 'U';
124*1858f998SYi Kong       else if (Diag == CblasNonUnit) DI = 'N';
125*1858f998SYi Kong       else
126*1858f998SYi Kong       {
127*1858f998SYi Kong          cblas_xerbla(4, "cblas_ztrmv","Illegal Diag setting, %d\n", Diag);
128*1858f998SYi Kong          CBLAS_CallFromC = 0;
129*1858f998SYi Kong          RowMajorStrg = 0;
130*1858f998SYi Kong          return;
131*1858f998SYi Kong       }
132*1858f998SYi Kong       #ifdef F77_CHAR
133*1858f998SYi Kong          F77_UL = C2F_CHAR(&UL);
134*1858f998SYi Kong          F77_TA = C2F_CHAR(&TA);
135*1858f998SYi Kong          F77_DI = C2F_CHAR(&DI);
136*1858f998SYi Kong       #endif
137*1858f998SYi Kong          F77_ztrmv( F77_UL, F77_TA, F77_DI, &F77_N, A, &F77_lda, X,
138*1858f998SYi Kong                       &F77_incX);
139*1858f998SYi Kong       if (TransA == CblasConjTrans)
140*1858f998SYi Kong       {
141*1858f998SYi Kong          if (N > 0)
142*1858f998SYi Kong          {
143*1858f998SYi Kong             do
144*1858f998SYi Kong             {
145*1858f998SYi Kong                *x = -(*x);
146*1858f998SYi Kong                x += i;
147*1858f998SYi Kong             }
148*1858f998SYi Kong             while (x != st);
149*1858f998SYi Kong          }
150*1858f998SYi Kong       }
151*1858f998SYi Kong    }
152*1858f998SYi Kong    else cblas_xerbla(1, "cblas_ztrmv", "Illegal Order setting, %d\n", order);
153*1858f998SYi Kong    CBLAS_CallFromC = 0;
154*1858f998SYi Kong    RowMajorStrg = 0;
155*1858f998SYi Kong    return;
156*1858f998SYi Kong }
157