tensor_node_executor.hpp

 
#ifndef EXATN_RUNTIME_TENSOR_NODE_EXECUTOR_HPP_
#define EXATN_RUNTIME_TENSOR_NODE_EXECUTOR_HPP_
 
#include "Identifiable.hpp"
 
#include "tensor_op_factory.hpp"
#include "tensor.hpp"
 
#include <vector>
#include <memory>
 
namespace talsh{
class Tensor;
}
 
namespace exatn {
namespace runtime {
 
class TensorNodeExecutor : public Identifiable, public Cloneable<TensorNodeExecutor> {
 
public:
 
  virtual ~TensorNodeExecutor() = default;
 
  virtual void initialize() = 0;
 
  virtual int execute(numerics::TensorOpCreate & op,
                      TensorOpExecHandle * exec_handle) = 0;
  virtual int execute(numerics::TensorOpDestroy & op,
                      TensorOpExecHandle * exec_handle) = 0;
  virtual int execute(numerics::TensorOpTransform & op,
                      TensorOpExecHandle * exec_handle) = 0;
  virtual int execute(numerics::TensorOpAdd & op,
                      TensorOpExecHandle * exec_handle) = 0;
  virtual int execute(numerics::TensorOpContract & op,
                      TensorOpExecHandle * exec_handle) = 0;
 
  virtual bool sync(TensorOpExecHandle op_handle,
                    int * error_code,
                    bool wait = false) = 0;
 
  virtual std::shared_ptr<talsh::Tensor> getLocalTensor(const numerics::Tensor & tensor,
                         const std::vector<std::pair<DimOffset,DimExtent>> & slice_spec) = 0;
 
  virtual std::shared_ptr<TensorNodeExecutor> clone() = 0;
};
 
} //namespace runtime
} //namespace exatn
 
#endif //EXATN_RUNTIME_TENSOR_NODE_EXECUTOR_HPP_