| 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( \
112 TRA, TRB, S1 / 100.0, S2 / 100.0); \
114 auto g = SparseMatMul<TA, TB>(M, N, K, S1 / 100.0, S2 / 100.0, TRA, TRB); \
118 BM_Sparse##_##M##_##K##_##N##_##S1##_##S2##_##TRA##_##TRB##_##TA##_##TB) \
135 #define BM_SPARSE_FLOAT(M, K, N, S1, S2, TRA, TRB) \ argument
136 BM_SPARSE(M, K, N, S1, S2, TRA, TRB, float, float)
137 #define BM_SPARSE_BFLOAT16(M, K, N, S1, S2, TRA, TRB) \ argument
138 BM_SPARSE(M, K, N, S1, S2, TRA, TRB, bfloat16, bfloat16)
139 #define BM_SPARSE_FLOAT_BFLOAT16(M, K, N, S1, S2, TRA, TRB) \ argument
[all …]
|