Lines Matching refs:BS
77 constexpr const int BS = 16; variable
83 for (int i = 0; i < N / BS; ++i) in BlockMatMul_TargetNowait()
84 for (int j = 0; j < N / BS; ++j) { in BlockMatMul_TargetNowait()
86 for (int k = 0; k < N / BS; ++k) { in BlockMatMul_TargetNowait()
91 map(to: BlockA[:BS * BS], BlockB[:BS * BS]) \ in BlockMatMul_TargetNowait()
92 map(tofrom: BlockC[:BS * BS]) nowait in BlockMatMul_TargetNowait()
95 for (int ii = 0; ii < BS; ii++) in BlockMatMul_TargetNowait()
96 for (int jj = 0; jj < BS; jj++) { in BlockMatMul_TargetNowait()
97 for (int kk = 0; kk < BS; ++kk) in BlockMatMul_TargetNowait()
98 BlockC[ii + jj * BS] += in BlockMatMul_TargetNowait()
99 BlockA[ii + kk * BS] * BlockB[kk + jj * BS]; in BlockMatMul_TargetNowait()
130 auto BlockedA = BlockMatrix(BS, BS, N, N); in main()
131 auto BlockedB = BlockMatrix(BS, BS, N, N); in main()
132 auto BlockedC = BlockMatrix(BS, BS, N, N); in main()