uTensor · Knight-X · Oct 29, 2017 · Oct 29, 2017 · Oct 29, 2017 · Oct 29, 2017
diff --git a/context.hpp b/context.hpp
@@ -0,0 +1,113 @@
+#ifndef UTENSOR_CTX_H
+#define UTENSOR_CTX_H
+
+#include <unordered_map>
+
+//TODO: how do we deal with dangling tensors?
+//      only allow pushing for exact number of inputs
+//      output reference count are initialized to 0, incremented only on input-push
+//      outputs are allocated in ops
+//      output lists can contain nullptr/empty-tensors
+//      tensors can be all pointers here, but destructors has to set data to nullptr
+//      push(op, input_t_list, output_t_list)  or  push(op, init-list, init-list)
+//      TensorListModifierOp
+class Context : uTensor {
+protected:
+  vector<Operator> op_list;
+  std::unordered_map<TensorBase*, int> tensor_refs;
+
+  void initOpTensors(vector<TensorBase*> &t_list);
+  void deinitTensors(vector<TensorBase*> &t_list);
+  void registerInputTensors(vector<TensorBase*> &t_list);
+  void registerOutputTensors(vector<TensorBase*> &t_list);
+  void decreRefCount(vector<TensorBase*> &t_list);
+
+  //void unref2nullTensors(vector<TensorBase*> &t_list);
+  //replace non-referenced output to null-tensors
+
+public:
+  void push(Operator op, TList _inputs, TList _outputs);
+  int run(void);
+};
+
+void push(Operator op, TList _inputs, TList _outputs) {
+  if(op.getInputCount() != _inputs.size()) {
+    ERR_EXIT("valid number of inputs\r\n");
+  }
+  if(op.getOutputCount() != _outputs.size()) {
+    ERR_EXIT("valid number of output\r\n");
+  }
+
+  op_list.push_back(op);
+  registerInputTensors(_inputs);
+  registerOutputTensors(_outputs);
+
+}
+
+
+void Context::registerInputTensors(TList &t_list) {
+  for(auto t:t_list) {
+    auto ref_count = tensor_refs.find(t);
+    if(ref_count == tensor_refs.end()) {
+      tensor_refs[t] = 1;
+    } else {
+      tensor_refs[t]++;
+    }
+  }
+}
+
+void Context::registerOutputTensors(TList &t_list) {
+  for(auto t:t_list) {
+    auto ref_count = tensor_refs.find(t);
+    if(ref_count == tensor_refs.end()) {
+      tensor_refs[t] = 0;
+    }
+  }
+}
+
+
+void Context::initOpTensors(vector<TensorBase*> &t_list) {
+  for(auto t:t_list) {
+    t.inFocus();
+  }
+}
+
+void Context::deinitTensors(vector<TensorBase*> &t_list) {
+  for(auto t:t_list) {
+    t.deFocus();
+  }
+}
+
+void Context::deinitTensors(vector<TensorBase*> &t_list) {
+  for(auto t:t_list) {
+    t.deFocus();
+  }
+}
+
+void Context::decreRefCount(vector<TensorBase*> &t_list) {
+  for(auto t:t_list) {
+    tensor_refs[t] = tensor_refs[t] - 1;
+    if(tensor_refs[t] < 1) {
+      t.~Tensor();
+    }
+}
+
+int Context::run(void) {
+  //unref2nullTensors();
+
+  for(auto op:op_list) {
+    initTensors(op.getInputs());
+    initTensors(op.getOutputs());
+
+    op.init();
+    op.compute();
+    op.deinit();
+
+    deinitOpTensors(op.getInputs());
+    deinitOpTensors(op.getOutputs());
+
+    decreRefCount(op.getInputs());
+  }
+}
+
+#endif // UTENSOR_CTX_H
diff --git a/uTensorBase.hpp b/uTensorBase.hpp
@@ -0,0 +1,64 @@
+#ifndef UTENSOR_BASE_H
+#define UTENSOR_BASE_H
+
+#include "tensor.hpp"
+
+typedef long long TensorPtr;
+typedef vector<TensorBase*> TList;
+
+class uTensor {
+  virtual void inFocus() {};
+  virtual void deFocus() {};
+  virtual ~uTensor() = 0;
+};
+
+
+//isType() https://stackoverflow.com/questions/9974596/how-to-check-whether-two-pointers-point-to-the-same-object-or-not
+//double dispatch
+
+//new vs stack
+class Operator {
+protected:
+  //setup input/output info in derived constructors
+  TList inputs;
+  vector<DType> dtype_in;
+  TList outputs;
+  vector<DType> dtype_out;
+public:
+  virtual void compute() = 0;
+
+  void setInputs(TList &_inputs) {
+    if(_inputs.size() != inputs.size()) ERR_EXIT("Input Tensor list mismatched...");
+
+    for(uint8_t i = 0; i < input.size(); i++) {
+      if(dtype_in[i] != inputs.getType()) {
+        ERR_EXIT("Tensor Type mismatched...");
+      }
+
+      input[i] = _inputs[i];
+    }
+  }
+
+  void setOutputs(TList &_outputs) {
+    if(_outputs.size() != outputs.size()) ERR_EXIT("Input Tensor list mismatched...");
+
+    for(uint8_t i = 0; i < output.size(); i++) {
+      if(dtype_out[i].getType() != output[i].getType()) {
+        ERR_EXIT("Tensor Type mismatched...");
+      }
+
+      output[i] = _output[i]
+    }
+  }
+
+  TList getInputs(void) {
+    return inputs;
+  }
+
+  TList getOutputs(void) {
+    return outputs;
+  }
+};
+
+
+#endif //UTENSOR_BASE_H