astra-sim-alibabacloud/astra-sim/system/AstraComputeAPI.hh

/******************************************************************************
This source code is licensed under the MIT license found in the
LICENSE file in the root directory of this source tree.
*******************************************************************************/

#ifndef __COMPUTE_HH__
#define __COMPUTE_HH__
#include <cassert>
#include <cstdint>
#include <fstream>
#include <iostream>
#include <string>
namespace AstraSim {
class ComputeMetaData {
 public:
  uint64_t compute_delay; 
};
class ComputeAPI {
 public:
  virtual void compute(
      uint64_t M,
      uint64_t K,
      uint64_t N,
      void (*msg_handler)(void* fun_arg),
      ComputeMetaData* fun_arg) = 0;
  virtual ~ComputeAPI() = 0;
};
} // namespace AstraSim
#endif