summaryrefslogtreecommitdiffstats
path: root/third_party/intgemm/intgemm/callbacks/output_buffer_info.h
blob: 213aef4afc704ea182eb952a7ff78f2975de48f8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#pragma once

#include "../types.h"

namespace intgemm {
namespace callbacks {

struct OutputBufferInfo {
  Index row_idx;
  Index col_idx;

  Index rows; // = A_rows
  Index cols; // = B_cols

  OutputBufferInfo(Index row_idx, Index col_idx, Index rows, Index cols)
    : row_idx(row_idx), col_idx(col_idx), rows(rows), cols(cols) {}
};

}
}