added tensorflow code generator

2019-08-15 15:59:53 -07:00
parent 38a8b0ab19
commit 3ece461ce2
4 changed files with 174 additions and 2 deletions
--- a/include/triton/runtime/function.h
+++ b/include/triton/runtime/function.h
@@ -97,6 +97,7 @@ public:
  function(const std::string& src);
  void operator()(const std::vector<arg>& args, const std::array<size_t, 3>& grid, driver::stream* stream);
  void operator()(const std::vector<arg>& args, const grid_fn_ty& grid, driver::stream *stream);
+  std::string make_tensorflow_src(const std::vector<size_t> &outputs, const std::string &macro);

 private:
  // execution context