fbgemm_pack_matrix_cache.h

#pragma once

#include "fbgemm/Fbgemm.h"

namespace caffe2 {

template <typename ACC_T>
std::shared_ptr<fbgemm::PackBMatrix<int8_t, ACC_T>>
GetOrCreateFbgemmPackBMatrix(
    fbgemm::matrix_op_t trans,
    std::int32_t m,
    std::int32_t n,
    const void* orig_data,
    const std::int8_t* quantized_data,
    std::int32_t ld);

} // namespace caffe2