Skip to content

Commit

Permalink
[uTVM] Initial BYOC support with c-source module
Browse files Browse the repository at this point in the history
* comment ext. -> external change
* reorder header c-source external demo backend.

Change-Id: I304aab904aa3fb19f4459f9d3b1bc3cd44e90710
  • Loading branch information
manupak committed Nov 24, 2020
1 parent 005708f commit 0480af2
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion include/tvm/ir/module.h
Original file line number Diff line number Diff line change
Expand Up @@ -56,7 +56,7 @@ class IRModuleNode : public Object {
Map<GlobalTypeVar, TypeData> type_definitions;
/*! \brief The source map for the module. */
parser::SourceMap source_map;
/*! \brief The names of ext. functions for func registry */
/*! \brief The names of external functions for func registry */
Array<String> ext_func_names;

IRModuleNode() : source_map() {}
Expand Down
6 changes: 3 additions & 3 deletions src/relay/backend/contrib/codegen_c/codegen.cc
Original file line number Diff line number Diff line change
Expand Up @@ -232,12 +232,12 @@ class CSourceCodegen : public CSourceModuleCodegenBase {
Array<String> variables = std::get<1>(res);

// Create headers
code_stream_ << "#include \"tvm/runtime/c_runtime_api.h\"\n";
code_stream_ << "#include \"tvm/runtime/c_backend_api.h\"\n";
code_stream_ << "#include <tvm/runtime/crt/module.h>\n";
code_stream_ << "#include <stdio.h>\n";
code_stream_ << "#include <stdlib.h>\n";
code_stream_ << "#include <string.h>\n";
code_stream_ << "#include <tvm/runtime/c_runtime_api.h>\n";
code_stream_ << "#include <tvm/runtime/c_backend_api.h>\n";
code_stream_ << "#include <tvm/runtime/crt/module.h>\n";
if (!variables.empty()) {
// These are only needed to handle metadata copying
code_stream_ << "#include <tvm/runtime/container.h>\n";
Expand Down

0 comments on commit 0480af2

Please sign in to comment.