Skip to content

T
tic
Commit 4072396e by Zhi Committed by Tianqi Chen
Options

fix codegenc (#4597)

parent fadea922
Showing with 2 additions and 2 deletions
src/relay/backend/contrib/codegen_c/codegen.cc
...@@ -147,7 +147,7 @@ class CSourceCodegen : public CSourceModuleCodegenBase { ...@@ -147,7 +147,7 @@ class CSourceCodegen : public CSourceModuleCodegenBase {
// Record the external symbol for runtime lookup. // Record the external symbol for runtime lookup.
auto sid = GetExtSymbol(func); auto sid = GetExtSymbol(func);
auto builder = CodegenC(sid); CodegenC builder(sid);
builder.VisitExpr(func->body); builder.VisitExpr(func->body);
code_stream_ << builder.JIT(); code_stream_ << builder.JIT();
} }
......
src/relay/backend/contrib/dnnl/codegen.cc
...@@ -238,7 +238,7 @@ class DNNLModuleCodegen : public CSourceModuleCodegenBase { ...@@ -238,7 +238,7 @@ class DNNLModuleCodegen : public CSourceModuleCodegenBase {
// Record the external symbol for runtime lookup. // Record the external symbol for runtime lookup.
auto sid = GetExtSymbol(func); auto sid = GetExtSymbol(func);
auto builder = CodegenDNNL(sid); CodegenDNNL builder(sid);
builder.VisitExpr(func->body); builder.VisitExpr(func->body);
code_stream_ << builder.JIT(); code_stream_ << builder.JIT();
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment