Home
last modified time | relevance | path

Searched refs:kSelectCols (Results 1 – 1 of 1) sorted by relevance

/dports/www/firefox/firefox-99.0/third_party/intgemm/test/
H A Dmultiply_test.cc134 const int kSelectCols = 24; in TestSelectColumnsB() local
135 Index select_cols[kSelectCols]; in TestSelectColumnsB()
141 AlignedVector<Integer> test(rows * kSelectCols); in TestSelectColumnsB()
142 …tine::SelectColumnsB(prepared.begin(), test.begin(), rows, select_cols, select_cols + kSelectCols); in TestSelectColumnsB()
145 AlignedVector<float> selected(rows * kSelectCols); in TestSelectColumnsB()
147 for (int c = 0; c < kSelectCols; ++c) { in TestSelectColumnsB()
148 assert(c + r * kSelectCols < rows * kSelectCols); in TestSelectColumnsB()
149 selected[c + r * kSelectCols] = input[select_cols[c] + r * cols]; in TestSelectColumnsB()
152 AlignedVector<Integer> ref(rows * kSelectCols); in TestSelectColumnsB()
153 Routine::PrepareB(selected.begin(), ref.begin(), 1, rows, kSelectCols); in TestSelectColumnsB()
[all …]