DO NOT SUBMIT, WIP: emitc with mlgo nuances

It's a hack

Author: mtrofin

llvm/lib/Analysis/models/saved-model-to-tflite.py

@@ -31,7 +31,10 @@ def main(argv):
     src_json = os.path.join(sm_dir, json_file)
     if tf.io.gfile.exists(src_json):
         tf.io.gfile.copy(src_json, os.path.join(tfl_dir, json_file))
-
+    tf.mlir.experimental.tflite_to_tosa_bytecode(
+        tfl_path,
+        os.path.join(tfl_dir, 'model.tosa')
+    )
 
 if __name__ == "__main__":
     main(sys.argv)

mlir/lib/Target/Cpp/TranslateToCpp.cpp

@@ -6,9 +6,11 @@
 //
 //===----------------------------------------------------------------------===//
 
+#include "mlir-c/BuiltinAttributes.h"
 #include "mlir/Dialect/ControlFlow/IR/ControlFlowOps.h"
 #include "mlir/Dialect/EmitC/IR/EmitC.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
+#include "mlir/IR/Attributes.h"
 #include "mlir/IR/BuiltinOps.h"
 #include "mlir/IR/BuiltinTypes.h"
 #include "mlir/IR/Dialect.h"
@@ -1137,8 +1139,35 @@ static LogicalResult printOperation(CppEmitter &emitter,
         "with multiple blocks needs variables declared at top");
   }
 
-  CppEmitter::Scope scope(emitter);
+  CppEmitter::Scope classScope(emitter);
   raw_indented_ostream &os = emitter.ostream();
+  os << "class MyClass final {\n";
+  auto argAttrs = functionOp.getArgAttrs();
+  
+  std::map<std::string, Value> fields;
+  if (argAttrs)
+    for (auto [a,v] : zip(*argAttrs, functionOp.getArguments())) {
+      if (auto da = dyn_cast<mlir::DictionaryAttr>(a)) {
+        auto name = cast<mlir::StringAttr>(
+          da.getNamed("ml_program.identifier")->getValue())
+          .str();
+        fields[name] = v;
+        os << "  ";
+        if (failed(emitter.emitType(functionOp.getLoc(), v.getType())))
+          return failure();
+        os << " " << emitter.getOrCreateName(v) << ";\n";
+      }
+    }
+  os << "  std::map<std::string, char*> _buffer_map {";
+  for (auto &[n,v]:fields)
+    os << "{ \"" << n << "\"" << ", reinterpret_cast<char*>(" << emitter.getOrCreateName(v) << ") },";
+  os << "  };\n";
+  os << "  char* getBufferForName(const std::string& name) const {\n";
+  os << "    auto it = _buffer_map.find(name);\n";
+  os << "    return (it == _buffer_map.end()) ? nullptr : it->second;\n";
+  os << "  }\n";
+  CppEmitter::Scope scope(emitter);
+  
   if (functionOp.getSpecifiers()) {
     for (Attribute specifier : functionOp.getSpecifiersAttr()) {
       os << cast<StringAttr>(specifier).str() << " ";
@@ -1159,13 +1188,13 @@ static LogicalResult printOperation(CppEmitter &emitter,
     os << ");";
     return success();
   }
-  if (failed(printFunctionArgs(emitter, operation, functionOp.getArguments())))
-    return failure();
+  // if (failed(printFunctionArgs(emitter, operation, functionOp.getArguments())))
+  //   return failure();
   os << ") {\n";
   if (failed(printFunctionBody(emitter, operation, functionOp.getBlocks())))
     return failure();
   os << "}\n";
-
+  os << "};\n";
   return success();
 }