@@ -211,12 +211,7 @@ int main() {
211
211
// A
212
212
test_get_coord_op<bfloat16, float , /* TM*/ 16 , /* TK*/ 16 , use::a,
213
213
layout::row_major, 1 >();
214
- // B
215
- // Duplicate from 8x16x16
216
- // test_get_coord_op<bfloat16, float, /*TK*/ 16, /*TN*/ 16, use::b,
217
- // layout::ext_intel_packed, 2>();
218
- // test_get_coord_op<bfloat16, float, /*TK*/ 16, /*TN*/ 16, use::b,
219
- // layout::row_major, 1>();
214
+ // B - duplicate from 8x16x16
220
215
// Accumulator
221
216
test_get_coord_op<bfloat16, float , /* TM*/ 16 , /* TN*/ 16 , use::accumulator,
222
217
layout::row_major, 1 >();
@@ -260,29 +255,19 @@ int main() {
260
255
// A
261
256
test_get_coord_op<bfloat16, float , /* TM*/ 32 , /* TK*/ 16 , use::a,
262
257
layout::row_major, 1 >();
263
- // B
264
- // Duplicate from 1x64x16
265
- // test_get_coord_op<bfloat16, float, /*TK*/ 16, /*TN*/ 64, use::b,
266
- // layout::ext_intel_packed, 2>();
267
- // test_get_coord_op<bfloat16, float, /*TK*/ 16, /*TN*/ 64, use::b,
268
- // layout::row_major, 1>();
258
+ // B - duplicate from 1x64x16
269
259
// Accumulator
270
260
test_get_coord_op<bfloat16, float , /* TM*/ 32 , /* TN*/ 64 , use::accumulator,
271
261
layout::row_major, 1 >();
272
262
test_get_coord_op<float , float , /* TM*/ 32 , /* TN*/ 64 , use::accumulator,
273
263
layout::row_major, 1 >();
274
264
275
- // // 32x64x32 float/bfloat16
265
+ // 32x64x32 float/bfloat16
276
266
std::cout << " 32x64x32 float/bfloat16" << std::endl;
277
267
// A
278
268
test_get_coord_op<bfloat16, float , /* TM*/ 32 , /* TK*/ 32 , use::a,
279
269
layout::row_major, 1 >();
280
- // B
281
- // Duplicate from 1x64x32
282
- // test_get_coord_op<bfloat16, float, /*TK*/ 32, /*TN*/ 64, use::b,
283
- // layout::ext_intel_packed, 2>();
284
- // test_get_coord_op<bfloat16, float, /*TK*/ 32, /*TN*/ 64, use::b,
285
- // layout::row_major, 1>();
270
+ // B - duplicate from 1x64x32
286
271
// Accumulator
287
272
test_get_coord_op<bfloat16, float , /* TM*/ 32 , /* TN*/ 64 , use::accumulator,
288
273
layout::row_major, 1 >();
0 commit comments