| 106 | } |
| 107 | |
| 108 | void init(int32_t m_, int32_t n_, int32_t k_, bool tA, bool tB) noexcept |
| 109 | { |
| 110 | m = m_; |
| 111 | n = n_; |
| 112 | k = k_; |
| 113 | transA = tA; |
| 114 | transB = tB; |
| 115 | ldA = transA ? k : m; |
| 116 | ldB = transB ? n : k; |
| 117 | ldC = m; |
| 118 | |
| 119 | rA = ldA; |
| 120 | rB = ldB; |
| 121 | rC = ldC; |
| 122 | |
| 123 | cA = transA ? m : k; |
| 124 | cB = transB ? k : n; |
| 125 | cC = n; |
| 126 | |
| 127 | opA = transA ? CUBLAS_OP_T : CUBLAS_OP_N; |
| 128 | opB = transB ? CUBLAS_OP_T : CUBLAS_OP_N; |
| 129 | |
| 130 | elemA = m * k; |
| 131 | elemB = n * k; |
| 132 | elemC = n * m; |
| 133 | bytesA = word_size * elemA; |
| 134 | bytesB = word_size * elemB; |
| 135 | bytesC = word_size * elemC; |
| 136 | alpha = T(1.f); |
| 137 | beta = T(0.f); |
| 138 | } |
| 139 | }; |
| 140 | |
| 141 | auto constexpr kNB_ALGO_COMBINATIONS = 6000; |