| @@ -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; | |||||
| } | |||||