Skip to content

Commit 743a6bf

Browse files
committed
Remove codegen hack
1 parent a79fe37 commit 743a6bf

2 files changed

Lines changed: 2 additions & 2 deletions

File tree

src/relay/backend/aot_executor_codegen.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -355,7 +355,7 @@ class AOTExecutorCodegen : public MixedModeVisitor {
355355
// DataType::Handle(). Need to fix this.
356356
auto param_handle = tvm::tir::Call(DataType::Handle(), tvm::tir::builtin::lookup_param(),
357357
{tir::StringImm(params_by_expr_[arg])});
358-
args.push_back(tvm::tir::Cast(DataType::Handle(), param_handle));
358+
args.push_back(tvm::tir::Cast(DataType::Handle(32, 1), param_handle));
359359
} else {
360360
auto sids = FindExpr(arg);
361361
PushArgs(arg, sids, &args);

src/target/source/codegen_c.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ void CodeGenC::PrintVecStore(const VarNode* buffer, DataType t, PrimExpr base,
345345
}
346346

347347
std::string CodeGenC::CastFromTo(std::string value, DataType from, DataType target) {
348-
// if (from == target) return value;
348+
if (from == target) return value;
349349
std::ostringstream os;
350350
os << "((";
351351
this->PrintType(target, os);

0 commit comments

Comments
 (0)