| D | sparse_matmul_op_test.cc | 107 #define BM_SPARSE(M, K, N, S1, S2, TRA, TRB, TA, TB) \ argument
109 BM_Sparse##_##M##_##K##_##N##_##S1##_##S2##_##TRA##_##TRB##_##TA##_##TB( \
114 TRA, TRB, S1 / 100.0, S2 / 100.0); \
117 auto g = SparseMatMul<TA, TB>(M, N, K, S1 / 100.0, S2 / 100.0, TRA, TRB); \
122 BM_Sparse##_##M##_##K##_##N##_##S1##_##S2##_##TRA##_##TRB##_##TA##_##TB);
141 #define BM_SPARSE_FLOAT(M, K, N, S1, S2, TRA, TRB) \ argument
142 BM_SPARSE(M, K, N, S1, S2, TRA, TRB, float, float)
143 #define BM_SPARSE_BFLOAT16(M, K, N, S1, S2, TRA, TRB) \ argument
144 BM_SPARSE(M, K, N, S1, S2, TRA, TRB, bfloat16, bfloat16)
145 #define BM_SPARSE_FLOAT_BFLOAT16(M, K, N, S1, S2, TRA, TRB) \ argument
[all …]
|