From 35f26361967c41dec909bb4cf664e5680daeaf89 Mon Sep 17 00:00:00 2001 From: Lily Orth-Smith Date: Tue, 2 Nov 2021 17:28:25 -0700 Subject: [PATCH] lint --- src/relay/backend/aot_executor_codegen.cc | 2 +- src/relay/backend/te_compiler.cc | 6 ++++-- src/relay/op/call/call.cc | 6 +++--- 3 files changed, 8 insertions(+), 6 deletions(-) diff --git a/src/relay/backend/aot_executor_codegen.cc b/src/relay/backend/aot_executor_codegen.cc index 173b1aa243cd0..9c3f27cca7cce 100644 --- a/src/relay/backend/aot_executor_codegen.cc +++ b/src/relay/backend/aot_executor_codegen.cc @@ -320,8 +320,8 @@ class AOTExecutorCodegen : public MixedModeVisitor { tvm::Array args{tvm::tir::StringImm(func_name)}; std::vector create_func_call_stmts; - // Pack the inputs + // Pack the inputs for (Expr arg : call_args) { if (params_by_expr_.find(arg) != params_by_expr_.end()) { auto param_handle = tvm::tir::Call(DataType::Handle(), tvm::tir::builtin::lookup_param(), diff --git a/src/relay/backend/te_compiler.cc b/src/relay/backend/te_compiler.cc index 04c6f343ca190..973773c2eb3bc 100644 --- a/src/relay/backend/te_compiler.cc +++ b/src/relay/backend/te_compiler.cc @@ -593,7 +593,8 @@ class LowerTensorExprMutator : public DeviceAwareExprMutator { } Expr DeviceAwareVisitExpr_(const CallNode* call_node) override { - ICHECK(call_node->op != CallLoweredOp()) << "Found a call_lowered op: can't lower a function that is already lowered."; + ICHECK(call_node->op != CallLoweredOp()) + << "Found a call_lowered op: can't lower a function that is already lowered."; Call call = GetRef(call_node); // Look for (indirect) calls to primitives. BaseFunc prim_func = ResolveToPrimitive(call_node->op); @@ -633,7 +634,8 @@ class LowerTensorExprMutator : public DeviceAwareExprMutator { for (const auto& arg : call_node->args) { args.push_back(VisitExpr(arg)); } - return CallLowered(pair.first, Tuple(args), Attrs(call_lowered_attrs), call_node->type_args, call_node->span); + return CallLowered(pair.first, Tuple(args), Attrs(call_lowered_attrs), call_node->type_args, + call_node->span); } IRModule module_; diff --git a/src/relay/op/call/call.cc b/src/relay/op/call/call.cc index 6615f94d36b8d..84a11ce8a0de0 100644 --- a/src/relay/op/call/call.cc +++ b/src/relay/op/call/call.cc @@ -62,12 +62,12 @@ Expr CallLowered(Expr func, Expr inputs, Attrs attrs, Array type_args, Spa // function. ICHECK(func.as()) << "Function to call should be GlobalVarNode, but got " << func->GetTypeKey(); - ICHECK(inputs.as()) - << "Inputs to call_lowered should be TupleNode, but got " << inputs->GetTypeKey(); + ICHECK(inputs.as()) << "Inputs to call_lowered should be TupleNode, but got " + << inputs->GetTypeKey(); return Call(CallLoweredOp(), {func, inputs}, attrs); } -TVM_REGISTER_GLOBAL("relay.op.call_lowered").set_body_typed(CallLowered); // TODO(@electriclilies): Not sure if type args and span are OK for registered op? +TVM_REGISTER_GLOBAL("relay.op.call_lowered").set_body_typed(CallLowered); RELAY_REGISTER_OP("call_lowered") .describe(R"code(Invoke an operation compiled by TVM.)code" TVM_ADD_FILELINE)