/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/tests/python/contrib/ |
H A D | test_rocblas.py | 53 bshape = (batch, n, k) if transb else (batch, k, n) 56 C = lib.batch_matmul(A, B, transa, transb) 59 def get_numpy(a, b, transa, transb): argument 62 if not transb: 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/blas/3/gemm/front/flamec/ |
H A D | FLA_Gemm_internal.c | 29 transb, in FLA_Gemm_internal() 57 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Gemm_internal() 59 else if ( transb == FLA_TRANSPOSE ) in FLA_Gemm_internal() 63 else if ( transb == FLA_CONJ_TRANSPOSE ) in FLA_Gemm_internal() 68 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Gemm_internal() 70 else if ( transb == FLA_TRANSPOSE ) in FLA_Gemm_internal() 74 else if ( transb == FLA_CONJ_TRANSPOSE ) in FLA_Gemm_internal() 79 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Gemm_internal() 81 else if ( transb == FLA_TRANSPOSE ) in FLA_Gemm_internal() 90 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Gemm_internal() [all …]
|
/dports/science/py-chainer/chainer-7.8.0/chainer/functions/math/ |
H A D | matmul.py | 43 transa, transb = not transb, not transa 47 if transb and b.ndim != 1: 77 self.transb = transb 168 if self.transb: 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) 240 self.transb = transb 277 self.transb = transb [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() 80 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_s8s8s32_pack_get_size() 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 66 bool is_packed_b = utils::one_of(*transb, 'P', 'p'); in check_gemm_input() 68 bool is_trans_b = utils::one_of(*transb, 'T', 't'); in check_gemm_input() 92 dnnl_status_t extended_sgemm(const char *transa, const char *transb, in extended_sgemm() argument 103 && utils::one_of(*transb, 'n', 'N', 't', 'T')) { in extended_sgemm() 105 bool trB = *transb == 't' || *transb == 'T'; in extended_sgemm() 157 bool BisN = (*transb == 'N' || *transb == '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 …]
|
H A D | gemm_pack.hpp | 33 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 38 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 43 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 48 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 53 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 57 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 62 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 67 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 71 dnnl_status_t DNNL_API sgemm_compute(const char *transa, const char *transb, 77 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, [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() 80 identifier, transa, transb, M, N, K, lda, ldb, size, pack); in gemm_s8s8s32_pack_get_size() 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 66 bool is_packed_b = utils::one_of(*transb, 'P', 'p'); in check_gemm_input() 68 bool is_trans_b = utils::one_of(*transb, 'T', 't'); in check_gemm_input() 92 dnnl_status_t extended_sgemm(const char *transa, const char *transb, in extended_sgemm() argument 103 && utils::one_of(*transb, 'n', 'N', 't', 'T')) { in extended_sgemm() 105 bool trB = *transb == 't' || *transb == 'T'; in extended_sgemm() 157 bool BisN = (*transb == 'N' || *transb == '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 …]
|
H A D | gemm_pack.hpp | 33 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 38 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 43 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 48 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 53 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 57 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 62 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 67 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 71 dnnl_status_t DNNL_API sgemm_compute(const char *transa, const char *transb, 77 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/gemm/ |
H A D | gemm_pack.cpp | 113 bool is_transb = utils::one_of(*transb, 'T', 't'); in check_pack_get_size_input() 116 && utils::one_of(*transb, '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() 415 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute() argument [all …]
|
H A D | gemm_pack.hpp | 34 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 38 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 43 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 48 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 53 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 57 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 62 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 66 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 69 dnnl_status_t sgemm_compute(const char *transa, const char *transb, 79 dnnl_status_t gemm_s8u8s32_compute(const char *transa, const char *transb, [all …]
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/gemm/ |
H A D | gemm_pack.cpp | 113 bool is_transb = utils::one_of(*transb, 'T', 't'); in check_pack_get_size_input() 116 && utils::one_of(*transb, '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() 415 dnnl_status_t sgemm_compute(const char *transa, const char *transb, in sgemm_compute() argument [all …]
|
H A D | gemm_pack.hpp | 34 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 38 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 43 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 48 const char *transa, const char *transb, const dim_t *M, const dim_t *N, 53 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 57 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 62 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 66 const char *transb, const dim_t *M, const dim_t *N, const dim_t *K, 69 dnnl_status_t sgemm_compute(const char *transa, const char *transb, 79 dnnl_status_t gemm_s8u8s32_compute(const char *transa, const char *transb, [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 167 transb = -1; 179 if (transB == 'N') transb = 0; 180 if (transB == 'T') transb = 1; 182 if (transB == 'R') transb = 0; 183 if (transB == 'C') transb = 1; 185 if (transB == 'R') transb = 2; 186 if (transB == 'C') transb = 3; 192 if (transb & 1) nrowb = args.n; 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 150 transb = -1; 162 if (transB == 'N') transb = 0; 163 if (transB == 'T') transb = 1; 165 if (transB == 'R') transb = 0; 166 if (transB == 'C') transb = 1; 168 if (transB == 'R') transb = 2; 169 if (transB == 'C') transb = 3; 175 if (transb & 1) nrowb = args.n; 212 int transa, transb; [all …]
|
/dports/math/gotoblas/GotoBLAS2/interface/ |
H A D | gemm.c | 112 int transa, transb, nrowa, nrowb; local 167 transb = -1; 179 if (transB == 'N') transb = 0; 180 if (transB == 'T') transb = 1; 182 if (transB == 'R') transb = 0; 183 if (transB == 'C') transb = 1; 185 if (transB == 'R') transb = 2; 186 if (transB == 'C') transb = 3; 192 if (transb & 1) nrowb = args.n; 229 int transa, transb; [all …]
|
/dports/math/openblas/OpenBLAS-0.3.18/interface/ |
H A D | gemm.c | 169 int transa, transb, nrowa, nrowb; local 227 transb = -1; 239 if (transB == 'N') transb = 0; 240 if (transB == 'T') transb = 1; 242 if (transB == 'R') transb = 0; 243 if (transB == 'C') transb = 1; 245 if (transB == 'R') transb = 2; 246 if (transB == 'C') transb = 3; 252 if (transb & 1) nrowb = args.n; 294 int transa, transb; [all …]
|
/dports/math/libflame/libflame-5.2.0/src/base/flamec/blis/3/ |
H A D | bl1_gemm.c | 57 bl1_screate_contigmt( transb, in bl1_sgemm() 105 bl1_toggle_trans( transb ); in bl1_sgemm() 175 bl1_toggle_trans( transb ); in bl1_sgemm() 217 transb, in bl1_sgemm() 249 transb, in bl1_sgemm() 478 transb, in bl1_dgemm() 510 transb, in bl1_dgemm() 810 transb, in bl1_cgemm() 842 transb, in bl1_cgemm() 1148 transb, in bl1_zgemm() [all …]
|
/dports/math/libflame/libflame-5.2.0/src/lapack/sol/sylv/front/flamec/ |
H A D | FLA_Sylv_internal.c | 16 FLA_Error FLA_Sylv_internal( FLA_Trans transa, FLA_Trans transb, FLA_Obj isgn, FLA_Obj A, FLA_Obj B… in FLA_Sylv_internal() argument 21 FLA_Sylv_internal_check( transa, transb, isgn, A, B, C, scale, cntl ); in FLA_Sylv_internal() 29 transb, in FLA_Sylv_internal() 42 ENQUEUE_FLASH_Sylv( transa, transb, isgn, A, B, C, scale, cntl ); in FLA_Sylv_internal() 57 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Sylv_internal() 59 else if ( transb == FLA_TRANSPOSE || transb == FLA_CONJ_TRANSPOSE ) in FLA_Sylv_internal() 64 if ( transb == FLA_NO_TRANSPOSE ) in FLA_Sylv_internal() 66 else if ( transb == FLA_TRANSPOSE || transb == FLA_CONJ_TRANSPOSE ) in FLA_Sylv_internal()
|
/dports/science/qmcpack/qmcpack-3.11.0/src/Platforms/CUDA/ |
H A D | cuBLAS.hpp | 81 const cublasOperation_t& transb, in gemm() argument 94 return cublasSgemm(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc); in gemm() 99 const cublasOperation_t& transb, in gemm() argument 112 …return cublasCgemm(handle, transa, transb, m, n, k, (const cuComplex*)alpha, (const cuComplex*)A, … in gemm() 118 const cublasOperation_t& transb, in gemm() argument 131 return cublasDgemm(handle, transa, transb, m, n, k, alpha, A, lda, B, ldb, beta, C, ldc); in gemm() 136 const cublasOperation_t& transb, in gemm() argument 155 const cublasOperation_t& transb, in gemm_batched() argument 174 const cublasOperation_t& transb, in gemm_batched() argument 194 const cublasOperation_t& transb, in gemm_batched() argument [all …]
|
/dports/devel/ga/ga-5.8/LinAlg/lapack+blas/ |
H A D | xgemm.c | 144 if (*transb == 'n' || *transb == 'N') in xb_sgemm() 173 if (*transb == 'n' || *transb == 'N') in xb_sgemm() 208 if (*transb == 'n' || *transb == 'N') in xb_sgemm() 237 if (*transb == 'n' || *transb == 'N') in xb_sgemm() 534 if (*transb == 'n' || *transb == 'N') in xb_dgemm() 563 if (*transb == 'n' || *transb == 'N') in xb_dgemm() 599 if (*transb == 'n' || *transb == 'N') in xb_dgemm() 628 if (*transb == 'n' || *transb == 'N') in xb_dgemm() 916 if (*transb == 'n' || *transb == 'N') in xb_zgemm() 945 if (*transb == 'n' || *transb == 'N') in xb_zgemm() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/python/tvm/contrib/ |
H A D | mkl.py | 22 def matmul(lhs, rhs, transa=False, transb=False, **kwargs): argument 43 m = rhs.shape[0] if transb else rhs.shape[1] 48 "tvm.contrib.mkl.matmul", ins[0], ins[1], outs[0], transa, transb 55 def matmul_u8s8s32(lhs, rhs, transa=False, transb=False, **kwargs): argument 76 m = rhs.shape[0] if transb else rhs.shape[1] 81 "tvm.contrib.mkl.matmul_u8s8s32", ins[0], ins[1], outs[0], transa, transb 88 def batch_matmul(lhs, rhs, transa=False, transb=False, iterative=False, **kwargs): argument 110 m = rhs.shape[1] if transb else rhs.shape[2] 122 transb,
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
H A D | gemm.cpp | 45 dnnl_status_t dnnl_sgemm(char transa, char transb, dim_t M, dim_t N, dim_t K, in dnnl_sgemm() argument 49 return cpu::extended_sgemm(&transb, &transa, &N, &M, &K, &alpha, B, &ldb, A, in dnnl_sgemm() 61 return cpu::gemm_s8x8s32(&transb, &transa, c2f_offsetC(&offsetc), &N, &M, in dnnl_gemm_u8s8s32() 73 return cpu::gemm_s8x8s32<int8_t>(&transb, &transa, c2f_offsetC(&offsetc), in dnnl_gemm_s8s8s32() 81 char transb, dim_t M, dim_t N, dim_t K, float alpha, in dnnl_gemm_bf16bf16f32() argument 93 dnnl_status_t dnnl_threadpool_interop_sgemm(char transa, char transb, dim_t M, in dnnl_threadpool_interop_sgemm() argument 105 dnnl_status_t dnnl_threadpool_interop_gemm_u8s8s32(char transa, char transb, in dnnl_threadpool_interop_gemm_u8s8s32() argument 117 dnnl_status_t dnnl_threadpool_interop_gemm_s8s8s32(char transa, char transb, in dnnl_threadpool_interop_gemm_s8s8s32() argument 123 status_t status = cpu::gemm_s8x8s32<int8_t>(&transb, &transa, in dnnl_threadpool_interop_gemm_s8s8s32() 131 char transa, char transb, dim_t M, dim_t N, dim_t K, float alpha, in dnnl_threadpool_interop_gemm_bf16bf16f32() argument [all …]
|
/dports/science/xtb/xtb-6.4.1/src/mctc/blas/ |
H A D | wrap3.f90 | 57 character(len=1), intent(in), optional :: transb local 82 character(len=1), intent(in), optional :: transb local 87 if (present(transb)) then 88 trb = transb 106 character(len=1), intent(in), optional :: transb local 116 if (present(transb)) then 117 trb = transb 170 if (present(transb)) then 171 trb = transb 200 if (present(transb)) then [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/tvm/src/runtime/contrib/cblas/ |
H A D | gemm_common.h | 72 bool transb = args[4]; in CallGemm() local 87 transb = IsInPlaceTransposed(B) ? !transb : transb; in CallGemm() 93 op(transb, transa, ColumnCount(B, transb), RowCount(A, transa), ColumnCount(A, transa), in CallGemm() 111 bool transb = args[4]; in CallU8S8S32Gemm() local 134 transb = IsInPlaceTransposed(B) ? !transb : transb; in CallU8S8S32Gemm() 141 op(transb, transa, ColumnCount(B, transb), RowCount(A, transa), ColumnCount(A, transa), in CallU8S8S32Gemm() 181 bool transb = args[4]; in CallBatchGemm() local 196 transb = IsInPlaceTransposed3D(B) ? !transb : transb; in CallBatchGemm() 210 op(batch_size, transb, transa, ColumnCount3D(B, transb), RowCount3D(A, transa), in CallBatchGemm()
|