| @@ -0,0 +1,171 @@ | |||
| /*********************************************************************/ | |||
| /* Copyright 2009, 2010 The University of Texas at Austin. */ | |||
| /* All rights reserved. */ | |||
| /* */ | |||
| /* 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 distribution. */ | |||
| /* */ | |||
| /* THIS SOFTWARE IS PROVIDED BY THE UNIVERSITY OF TEXAS AT */ | |||
| /* AUSTIN ``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 OF TEXAS AT */ | |||
| /* AUSTIN OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, */ | |||
| /* INCIDENTAL, SPECIAL, 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 THEORY OF */ | |||
| /* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT */ | |||
| /* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT */ | |||
| /* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE */ | |||
| /* POSSIBILITY OF SUCH DAMAGE. */ | |||
| /* */ | |||
| /* The views and conclusions contained in the software and */ | |||
| /* documentation are those of the authors and should not be */ | |||
| /* interpreted as representing official policies, either expressed */ | |||
| /* or implied, of The University of Texas at Austin. */ | |||
| /*********************************************************************/ | |||
| #include <stdio.h> | |||
| #include "common.h" | |||
| #ifdef FUNCTION_PROFILE | |||
| #include "functable.h" | |||
| #endif | |||
| #ifdef XDOUBLE | |||
| #define ERROR_NAME "XTRTRS" | |||
| #elif defined(DOUBLE) | |||
| #define ERROR_NAME "ZTRTRS" | |||
| #else | |||
| #define ERROR_NAME "CTRTRS" | |||
| #endif | |||
| static blasint (*trtrs_single[])(blas_arg_t *, BLASLONG *, BLASLONG *, FLOAT *, FLOAT *, BLASLONG) = { | |||
| TRTRS_UNU_SINGLE, TRTRS_UNN_SINGLE, TRTRS_UTU_SINGLE, TRTRS_UTN_SINGLE, TRTRS_URU_SINGLE, TRTRS_URN_SINGLE, TRTRS_UCU_SINGLE, TRTRS_UCN_SINGLE, TRTRS_LNU_SINGLE, TRTRS_LNN_SINGLE, TRTRS_LTU_SINGLE, TRTRS_LTN_SINGLE, TRTRS_LRU_SINGLE, TRTRS_LRN_SINGLE, TRTRS_LCU_SINGLE, TRTRS_LCN_SINGLE, | |||
| }; | |||
| #ifdef SMP | |||
| static blasint (*trtrs_parallel[])(blas_arg_t *, BLASLONG *, BLASLONG *, FLOAT *, FLOAT *, BLASLONG) = { | |||
| TRTRS_UNU_PARALLEL, TRTRS_UNN_PARALLEL, TRTRS_UTU_PARALLEL, TRTRS_UTN_PARALLEL, TRTRS_URU_PARALLEL, TRTRS_URN_PARALLEL, TRTRS_UCU_PARALLEL, TRTRS_UCN_PARALLEL, TRTRS_LNU_PARALLEL, TRTRS_LNN_PARALLEL, TRTRS_LTU_PARALLEL, TRTRS_LTN_PARALLEL, TRTRS_LRU_PARALLEL, TRTRS_LRN_PARALLEL, TRTRS_LCU_PARALLEL, TRTRS_LCN_PARALLEL, | |||
| }; | |||
| #endif | |||
| int NAME(char *UPLO, char* TRANS, char* DIAG, blasint *N, blasint *NRHS, FLOAT *a, blasint *ldA, | |||
| FLOAT *b, blasint *ldB, blasint *Info){ | |||
| char uplo_arg = *UPLO; | |||
| char trans_arg = *TRANS; | |||
| char diag_arg = *DIAG; | |||
| blas_arg_t args; | |||
| blasint info; | |||
| int uplo, trans, diag; | |||
| FLOAT *buffer; | |||
| #ifdef PPC440 | |||
| extern | |||
| #endif | |||
| FLOAT *sa, *sb; | |||
| PRINT_DEBUG_NAME; | |||
| args.m = *N; | |||
| args.n = *NRHS; | |||
| args.a = (void *)a; | |||
| args.lda = *ldA; | |||
| args.b = (void *)b; | |||
| args.ldb = *ldB; | |||
| info = 0; | |||
| TOUPPER(trans_arg); | |||
| trans = -1; | |||
| if (trans_arg == 'N') trans = 0; | |||
| if (trans_arg == 'T') trans = 1; | |||
| if (trans_arg == 'R') trans = 2; | |||
| if (trans_arg == 'C') trans = 3; | |||
| uplo = -1; | |||
| if (uplo_arg == 'U') uplo = 0; | |||
| if (uplo_arg == 'L') uplo = 1; | |||
| diag = -1; | |||
| if (diag_arg == 'U') diag = 0; | |||
| if (diag_arg == 'N') diag = 1; | |||
| if (args.ldb < MAX(1, args.m)) info = 7; | |||
| if (args.lda < MAX(1, args.m)) info = 9; | |||
| if (args.n < 0) info = 5; | |||
| if (args.m < 0) info = 4; | |||
| if (trans < 0) info = 2; | |||
| if (uplo < 0) info = 1; | |||
| if (diag < 0) info = 3; | |||
| if (info != 0) { | |||
| BLASFUNC(xerbla)(ERROR_NAME, &info, sizeof(ERROR_NAME)); | |||
| *Info = - info; | |||
| return 0; | |||
| } | |||
| args.alpha = NULL; | |||
| args.beta = NULL; | |||
| *Info = 0; | |||
| if (args.m == 0 || args.n == 0) return 0; | |||
| if (diag) { | |||
| if (AMIN_K(args.n, args.a, args.lda + 1) == ZERO) { | |||
| *Info = IAMIN_K(args.n, args.a, args.lda + 1); | |||
| return 0; | |||
| } | |||
| } | |||
| IDEBUG_START; | |||
| FUNCTION_PROFILE_START(); | |||
| #ifndef PPC440 | |||
| buffer = (FLOAT *)blas_memory_alloc(1); | |||
| sa = (FLOAT *)((BLASLONG)buffer + GEMM_OFFSET_A); | |||
| sb = (FLOAT *)(((BLASLONG)sa + ((GEMM_P * GEMM_Q * COMPSIZE * SIZE + GEMM_ALIGN) & ~GEMM_ALIGN)) + GEMM_OFFSET_B); | |||
| #endif | |||
| #ifdef SMP | |||
| args.common = NULL; | |||
| args.nthreads = num_cpu_avail(4); | |||
| if (args.nthreads == 1) { | |||
| #endif | |||
| (trtrs_single[(uplo << 2) | (trans << 1) | diag])(&args, NULL, NULL, sa, sb, 0); | |||
| #ifdef SMP | |||
| } else { | |||
| (trtrs_parallel[(uplo << 2) | (trans << 1) | diag])(&args, NULL, NULL, sa, sb, 0); | |||
| } | |||
| #endif | |||
| #ifndef PPC440 | |||
| blas_memory_free(buffer); | |||
| #endif | |||
| FUNCTION_PROFILE_END(COMPSIZE * COMPSIZE, args.m * args.n, 2 * args.m * args.m * args.n); | |||
| IDEBUG_END; | |||
| return 0; | |||
| } | |||