/dports/science/qmcpack/qmcpack-3.11.0/src/AFQMC/Numerics/detail/CPU/ |
H A D | sparse_cpu.hpp | 48 if (transa == 'n' || transa == 'N') in csrmv() 61 else if (transa == 't' || transa == 'T') in csrmv() 75 else if (transa == 'h' || transa == 'H' || transa == 'c' || transa == 'C') in csrmv() 108 if (transa == 'n' || transa == 'N') in csrmv() 121 else if (transa == 't' || transa == 'T') in csrmv() 135 else if (transa == 'h' || transa == 'H' || transa == 'c' || transa == 'C') in csrmv() 171 if (transa == 'n' || transa == 'N') in csrmm() 191 else if (transa == 't' || transa == 'T') in csrmm() 213 else if (transa == 'h' || transa == 'H' || transa == 'c' || transa == 'C') in csrmm() 257 if (transa == 'n' || transa == 'N') in csrmm() [all …]
|
/dports/math/libflame/libflame-5.2.0/src/blas/3/trsm/front/flamec/ |
H A D | FLA_Trsm_internal.c | 30 transa, in FLA_Trsm_internal() 59 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trsm_internal() 61 else if ( transa == FLA_TRANSPOSE ) in FLA_Trsm_internal() 65 else if ( transa == FLA_CONJ_TRANSPOSE ) in FLA_Trsm_internal() 70 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trsm_internal() 72 else if ( transa == FLA_TRANSPOSE ) in FLA_Trsm_internal() 76 else if ( transa == FLA_CONJ_TRANSPOSE ) in FLA_Trsm_internal() 84 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trsm_internal() 86 else if ( transa == FLA_TRANSPOSE ) in FLA_Trsm_internal() 95 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trsm_internal() [all …]
|
/dports/math/libflame/libflame-5.2.0/src/blas/3/trmm/front/flamec/ |
H A D | FLA_Trmm_internal.c | 30 transa, in FLA_Trmm_internal() 59 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trmm_internal() 61 else if ( transa == FLA_TRANSPOSE ) in FLA_Trmm_internal() 65 else if ( transa == FLA_CONJ_TRANSPOSE ) in FLA_Trmm_internal() 70 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trmm_internal() 72 else if ( transa == FLA_TRANSPOSE ) in FLA_Trmm_internal() 76 else if ( transa == FLA_CONJ_TRANSPOSE ) in FLA_Trmm_internal() 84 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trmm_internal() 86 else if ( transa == FLA_TRANSPOSE ) in FLA_Trmm_internal() 95 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trmm_internal() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/contrib/ |
H A D | test_rocblas.py | 52 ashape = (batch, k, m) if transa else (batch, m, k) 56 C = lib.batch_matmul(A, B, transa, transb) 59 def get_numpy(a, b, transa, transb): argument 60 if transa: 89 verify_batch_matmul(128, 64, 512, 512, rocblas, transa=False, transb=True) 90 verify_batch_matmul(128, 64, 512, 512, rocblas, transa=True, transb=False) 91 verify_batch_matmul(128, 64, 512, 512, rocblas, transa=True, transb=True) 95 verify_batch_matmul(128, 512, 512, 64, rocblas, transa=True, transb=True) 99 verify_batch_matmul(128, 512, 64, 512, rocblas, transa=True, transb=True) 103 verify_batch_matmul(128, 64, 128, 128, rocblas, transa=True, transb=True) [all …]
|
/dports/math/libflame/libflame-5.2.0/src/base/flamec/blis/2/ |
H A D | bl1_gemv.c | 25 if ( bl1_does_trans( transa ) ) n_elem = n; in bl1_sgemv() 52 bl1_toggle_trans( transa ); in bl1_sgemv() 55 bl1_sgemv_blas( transa, in bl1_sgemv() 108 bl1_toggle_trans( transa ); in bl1_dgemv() 111 bl1_dgemv_blas( transa, in bl1_dgemv() 171 bl1_toggle_trans( transa ); in bl1_cgemv() 197 if ( bl1_is_conjnotrans( transa ) ) in bl1_cgemv() 236 bl1_cgemv_blas( transa, in bl1_cgemv() 301 bl1_toggle_trans( transa ); in bl1_zgemv() 327 if ( bl1_is_conjnotrans( transa ) ) in bl1_zgemv() [all …]
|
/dports/science/py-chainer/chainer-7.8.0/chainer/functions/math/ |
H A D | matmul.py | 43 transa, transb = not transb, not transa 45 if transa and a.ndim != 1: 76 self.transa = transa 148 if self.transa: 187 def matmul(a, b, transa=False, transb=False): argument 220 return MatMul(transa=transa, transb=transb).apply((a, b))[0] 230 def _batch_matmul(a, b, transa, transb, transout): argument 233 return _matmul(a, b, transa, transb, transout) 239 self.transa = transa 276 self.transa = transa [all …]
|
/dports/math/xtensor-blas/xtensor-blas-0.20.0/include/xflens/cxxblas/drivers/ |
H A D | sparseblas.h | 27 void mkl_scscmv(const char *transa, 40 void mkl_scsrmv(const char *transa, 53 void mkl_dcscmv(const char *transa, 66 void mkl_dcsrmv(const char *transa, 79 void mkl_ccscmv(const char *transa, 92 void mkl_ccsrmv(const char *transa, 105 void mkl_zcscmv(const char *transa, 118 void mkl_zcsrmv(const char *transa, 132 void mkl_scsrsv(const char *transa, 143 void mkl_scscsv(const char *transa, [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/gemm/ |
H A D | gemm_pack.cpp | 47 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in sgemm_pack_get_size() 58 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_bf16bf16f32_pack_get_size() 69 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_s8u8s32_pack_get_size() 85 dnnl_status_t sgemm_pack(const char *identifier, const char *transa, in sgemm_pack() argument 90 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in sgemm_pack() 101 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_bf16bf16f32_pack() 111 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_s8u8s32_pack() 121 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_s8s8s32_pack() 126 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute() argument 132 transa, transb, M, N, K, A, lda, B, ldb, beta, C, ldc); in sgemm_compute() [all …]
|
H A D | gemm.cpp | 49 dnnl_status_t check_gemm_input(const char *transa, const char *transb, in check_gemm_input() argument 59 && utils::one_of(*transa, 'T', 't', 'N', 'n', 'P', 'p') in check_gemm_input() 65 bool is_packed_a = utils::one_of(*transa, 'P', 'p'); in check_gemm_input() 67 bool is_trans_a = utils::one_of(*transa, 'T', 't'); in check_gemm_input() 92 dnnl_status_t extended_sgemm(const char *transa, const char *transb, in extended_sgemm() argument 104 bool trA = *transa == 't' || *transa == 'T'; in extended_sgemm() 156 bool AisN = (*transa == 'N' || *transa == 'n'); in try_cblas_gemm_s8u8s32() 174 dnnl_status_t gemm_s8x8s32(const char *transa, const char *transb, in gemm_s8x8s32() argument 200 dnnl_status_t gemm_s8x8s32(const char *transa, const char *transb, in gemm_s8x8s32() argument 229 dnnl_status_t gemm_bf16bf16f32(const char *transa, const char *transb, in gemm_bf16bf16f32() argument [all …]
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/gemm/ |
H A D | gemm_pack.cpp | 47 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in sgemm_pack_get_size() 58 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_bf16bf16f32_pack_get_size() 69 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_s8u8s32_pack_get_size() 85 dnnl_status_t sgemm_pack(const char *identifier, const char *transa, in sgemm_pack() argument 90 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in sgemm_pack() 101 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_bf16bf16f32_pack() 111 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_s8u8s32_pack() 121 identifier, transa, transb, M, N, K, lda, ldb, src, dst); in gemm_s8s8s32_pack() 126 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute() argument 132 transa, transb, M, N, K, A, lda, B, ldb, beta, C, ldc); in sgemm_compute() [all …]
|
H A D | gemm.cpp | 49 dnnl_status_t check_gemm_input(const char *transa, const char *transb, in check_gemm_input() argument 59 && utils::one_of(*transa, 'T', 't', 'N', 'n', 'P', 'p') in check_gemm_input() 65 bool is_packed_a = utils::one_of(*transa, 'P', 'p'); in check_gemm_input() 67 bool is_trans_a = utils::one_of(*transa, 'T', 't'); in check_gemm_input() 92 dnnl_status_t extended_sgemm(const char *transa, const char *transb, in extended_sgemm() argument 104 bool trA = *transa == 't' || *transa == 'T'; in extended_sgemm() 156 bool AisN = (*transa == 'N' || *transa == 'n'); in try_cblas_gemm_s8u8s32() 174 dnnl_status_t gemm_s8x8s32(const char *transa, const char *transb, in gemm_s8x8s32() argument 200 dnnl_status_t gemm_s8x8s32(const char *transa, const char *transb, in gemm_s8x8s32() argument 229 dnnl_status_t gemm_bf16bf16f32(const char *transa, const char *transb, in gemm_bf16bf16f32() argument [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/ |
H A D | gemm_pack.cpp | 112 bool is_transa = utils::one_of(*transa, 'T', 't'); in check_pack_get_size_input() 115 bool ok = true && utils::one_of(*transa, 'T', 't', 'N', 'n') in check_pack_get_size_input() 134 identifier, transa, transb, M, N, K, lda, ldb); in check_pack_input() 174 identifier, transa, transb, M, N, K, lda, ldb); in sgemm_pack_get_size() 192 && utils::one_of(do_a ? *transa : *transb, 'n', 'N') in sgemm_pack_get_size() 212 identifier, transa, transb, M, N, K, lda, ldb); in gemm_bf16bf16f32_pack_get_size() 239 identifier, transa, transb, M, N, K, lda, ldb); in gemm_x8x8s32_pack_get_size() 317 auto trans = (cblas_id == CblasAMatrix) ? transa : transb; in sgemm_pack() 365 auto trans = (cblas_id == CblasAMatrix) ? transa : transb; in gemm_x8x8s32_pack() 423 cblas_sgemm_compute(CblasColMajor, cblas_storage(transa), in sgemm_compute() [all …]
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/ |
H A D | gemm_pack.cpp | 112 bool is_transa = utils::one_of(*transa, 'T', 't'); in check_pack_get_size_input() 115 bool ok = true && utils::one_of(*transa, 'T', 't', 'N', 'n') in check_pack_get_size_input() 134 identifier, transa, transb, M, N, K, lda, ldb); in check_pack_input() 174 identifier, transa, transb, M, N, K, lda, ldb); in sgemm_pack_get_size() 192 && utils::one_of(do_a ? *transa : *transb, 'n', 'N') in sgemm_pack_get_size() 212 identifier, transa, transb, M, N, K, lda, ldb); in gemm_bf16bf16f32_pack_get_size() 239 identifier, transa, transb, M, N, K, lda, ldb); in gemm_x8x8s32_pack_get_size() 317 auto trans = (cblas_id == CblasAMatrix) ? transa : transb; in sgemm_pack() 365 auto trans = (cblas_id == CblasAMatrix) ? transa : transb; in gemm_x8x8s32_pack() 423 cblas_sgemm_compute(CblasColMajor, cblas_storage(transa), in sgemm_compute() [all …]
|
/dports/multimedia/opentoonz/opentoonz-1.5.0/thirdparty/openblas/xianyi-OpenBLAS-e6e87a2/interface/ |
H A D | gemm.c | 112 int transa, transb, nrowa, nrowb; local 166 transa = -1; 169 if (transA == 'N') transa = 0; 170 if (transA == 'T') transa = 1; 172 if (transA == 'R') transa = 0; 173 if (transA == 'C') transa = 1; 175 if (transA == 'R') transa = 2; 176 if (transA == 'C') transa = 3; 190 if (transa & 1) nrowa = args.k; 229 int transa, transb; [all …]
|
/dports/cad/meshlab/meshlab-Meshlab-2020.05/src/plugins_unsupported/external/GotoBLAS2/interface/ |
H A D | gemm.c | 95 int transa, transb, nrowa, nrowb; local 149 transa = -1; 152 if (transA == 'N') transa = 0; 153 if (transA == 'T') transa = 1; 155 if (transA == 'R') transa = 0; 156 if (transA == 'C') transa = 1; 158 if (transA == 'R') transa = 2; 159 if (transA == 'C') transa = 3; 173 if (transa & 1) nrowa = args.k; 212 int transa, transb; [all …]
|
/dports/math/gotoblas/GotoBLAS2/interface/ |
H A D | gemm.c | 112 int transa, transb, nrowa, nrowb; local 166 transa = -1; 169 if (transA == 'N') transa = 0; 170 if (transA == 'T') transa = 1; 172 if (transA == 'R') transa = 0; 173 if (transA == 'C') transa = 1; 175 if (transA == 'R') transa = 2; 176 if (transA == 'C') transa = 3; 190 if (transa & 1) nrowa = args.k; 229 int transa, transb; [all …]
|
/dports/math/blasfeo/blasfeo-0.1.2/blas_api/ |
H A D | xtrsm_ref.c | 38 void TRSM(char *side, char *uplo, char *transa, char *diag, int *pm, int *pn, REAL *alpha, REAL *A,… in TRSM() argument 52 if( !(*transa=='c' | *transa=='C' | *transa=='n' | *transa=='N' | *transa=='t' | *transa=='T') ) in TRSM() 111 if(*transa=='n' | *transa=='N') // _lln in TRSM() 136 if(*transa=='n' | *transa=='N') // _lun in TRSM() 164 if(*transa=='n' | *transa=='N') // _rln in TRSM() 189 if(*transa=='n' | *transa=='N') // _run in TRSM()
|
/dports/math/openblas/OpenBLAS-0.3.18/interface/ |
H A D | gemm.c | 169 int transa, transb, nrowa, nrowb; local 226 transa = -1; 229 if (transA == 'N') transa = 0; 230 if (transA == 'T') transa = 1; 232 if (transA == 'R') transa = 0; 233 if (transA == 'C') transa = 1; 235 if (transA == 'R') transa = 2; 236 if (transA == 'C') transa = 3; 250 if (transa & 1) nrowa = args.k; 294 int transa, transb; [all …]
|
/dports/math/libxsmm/libxsmm-1.16.3/src/ |
H A D | libxsmm_spmdm.c | 174 char transa, in internal_spmdm_createSparseSlice_fp32_thread_sw() argument 189 char transa, in LIBXSMM_INTRINSICS() 209 char transa, in LIBXSMM_INTRINSICS() 229 char transa, in libxsmm_spmdm_createSparseSlice_fp32_thread() argument 251 char transa, in internal_spmdm_createSparseSlice_bfloat16_thread_sw() argument 266 char transa, in LIBXSMM_INTRINSICS() 286 char transa, in LIBXSMM_INTRINSICS() 306 char transa, in libxsmm_spmdm_createSparseSlice_bfloat16_thread() argument 328 char transa, in internal_spmdm_compute_fp32_thread_sw() argument 348 char transa, in LIBXSMM_INTRINSICS() [all …]
|
/dports/math/clblas/clBLAS-2.10/src/tests/correctness/ |
H A D | blas-lapack.c | 190 strmv_( &uplo, &transa, &diag, &n, a, &lda, x, &incx); in strmv() 196 dtrmv_( &uplo, &transa, &diag, &n, a, &lda, x, &incx); in dtrmv() 202 ctrmv_( &uplo, &transa, &diag, &n, a, &lda, x, &incx); in ctrmv() 358 stpmv_( &uplo, &transa, &diag, &n, ap, x, &incx); in stpmv() 364 dtpmv_( &uplo, &transa, &diag, &n, ap, x, &incx); in dtpmv() 370 ctpmv_( &uplo, &transa, &diag, &n, ap, x, &incx); in ctpmv() 376 ztpmv_( &uplo, &transa, &diag, &n, ap, x, &incx); in ztpmv() 382 stpsv_( &uplo, &transa, &diag, &n, ap, x, &incx); in stpsv() 388 dtpsv_( &uplo, &transa, &diag, &n, ap, x, &incx); in dtpsv() 394 ctpsv_( &uplo, &transa, &diag, &n, ap, x, &incx); in ctpsv() [all …]
|
/dports/math/libflame/libflame-5.2.0/src/base/flamec/blis/3/ |
H A D | bl1_gemm.c | 51 bl1_screate_contigmt( transa, in bl1_sgemm() 116 bl1_toggle_trans( transa ); in bl1_sgemm() 156 bl1_toggle_trans( transa ); in bl1_sgemm() 216 bl1_sgemm_blas( transa, in bl1_sgemm() 248 bl1_sgemm_blas( transa, in bl1_sgemm() 477 bl1_dgemm_blas( transa, in bl1_dgemm() 509 bl1_dgemm_blas( transa, in bl1_dgemm() 809 bl1_cgemm_blas( transa, in bl1_cgemm() 841 bl1_cgemm_blas( transa, in bl1_cgemm() 1147 bl1_zgemm_blas( transa, in bl1_zgemm() [all …]
|
/dports/math/libflame/libflame-5.2.0/src/blas/2/trsv/front/flamec/ |
H A D | FLA_Trsv_internal.c | 16 FLA_Error FLA_Trsv_internal( FLA_Uplo uplo, FLA_Trans transa, FLA_Diag diag, FLA_Obj A, FLA_Obj x, … in FLA_Trsv_internal() argument 21 FLA_Trsv_internal_check( uplo, transa, diag, A, x, cntl ); in FLA_Trsv_internal() 29 transa, in FLA_Trsv_internal() 40 ENQUEUE_FLASH_Trsv( uplo, transa, diag, A, x, cntl ); in FLA_Trsv_internal() 55 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trsv_internal() 57 else if ( transa == FLA_TRANSPOSE ) in FLA_Trsv_internal() 59 else if ( transa == FLA_CONJ_TRANSPOSE ) in FLA_Trsv_internal() 64 if ( transa == FLA_NO_TRANSPOSE ) in FLA_Trsv_internal() 66 else if ( transa == FLA_TRANSPOSE ) in FLA_Trsv_internal() 68 else if ( transa == FLA_CONJ_TRANSPOSE ) in FLA_Trsv_internal()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/cblas/ |
H A D | gemm_common.h | 71 bool transa = args[3]; in CallGemm() local 86 transa = IsInPlaceTransposed(A) ? !transa : transa; in CallGemm() 93 op(transb, transa, ColumnCount(B, transb), RowCount(A, transa), ColumnCount(A, transa), in CallGemm() 110 bool transa = args[3]; in CallU8S8S32Gemm() local 133 transa = IsInPlaceTransposed(A) ? !transa : transa; in CallU8S8S32Gemm() 141 op(transb, transa, ColumnCount(B, transb), RowCount(A, transa), ColumnCount(A, transa), in CallU8S8S32Gemm() 180 bool transa = args[3]; in CallBatchGemm() local 195 transa = IsInPlaceTransposed3D(A) ? !transa : transa; in CallBatchGemm() 210 op(batch_size, transb, transa, ColumnCount3D(B, transb), RowCount3D(A, transa), in CallBatchGemm() 211 ColumnCount3D(A, transa), static_cast<typename TBatchGemmOp::TDatatype>(alpha), B_data, B_size, in CallBatchGemm()
|
/dports/math/clblas/clBLAS-2.10/src/tests/include/ |
H A D | blas-cblas.h | 60 void strmv(char uplo, char transa, char diag, int n, float *a, int lda, float *x, int incx); 61 void dtrmv(char uplo, char transa, char diag, int n, double *a, int lda, double *x, int incx); 62 void ctrmv(char uplo, char transa, char diag, int n, complex *a, int lda, complex *x, int incx); 65 void strsv(char uplo, char transa, char diag, int n, float *a, int lda, float *x, int incx); 66 void dtrsv(char uplo, char transa, char diag, int n, double *a, int lda, double *x, int incx); 92 void stpmv(char uplo, char transa, char diag, int n, float *ap, float *x, int incx); 93 void dtpmv(char uplo, char transa, char diag, int n, double *ap, double *x, int incx); 94 void ctpmv(char uplo, char transa, char diag, int n, complex *ap, complex *x, int incx); 97 void stpsv(char uplo, char transa, char diag, int n, float *ap, float *x, int incx); 98 void dtpsv(char uplo, char transa, char diag, int n, double *ap, double *x, int incx); [all …]
|
/dports/science/qmcpack/qmcpack-3.11.0/src/Platforms/CUDA/ |
H A D | cuBLAS.hpp | 80 const cublasOperation_t& transa, in gemm() argument 94 return cublasSgemm(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc); in gemm() 98 const cublasOperation_t& transa, in gemm() argument 112 …return cublasCgemm(handle, transa, transb, m, n, k, (const cuComplex*)alpha, (const cuComplex*)A, … in gemm() 117 const cublasOperation_t& transa, in gemm() argument 131 return cublasDgemm(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc); in gemm() 135 const cublasOperation_t& transa, in gemm() argument 154 const cublasOperation_t& transa, in gemm_batched() argument 173 const cublasOperation_t& transa, in gemm_batched() argument 193 const cublasOperation_t& transa, in gemm_batched() argument [all …]
|