diff --git a/include/tvm/auto_scheduler/compute_dag.h b/include/tvm/auto_scheduler/compute_dag.h index b9c1f9e8c45c..71652fd692fa 100644 --- a/include/tvm/auto_scheduler/compute_dag.h +++ b/include/tvm/auto_scheduler/compute_dag.h @@ -68,7 +68,8 @@ class AccessAnalyzerNode : public Object { * (e.g., injective, broadcast and elementwise ops without reduction) */ OperationMap is_simple_access; /*! \brief Store whether the operation is strictly-inlineable - * (e.g., injective, broadcast and elementwise without reduction, branch or expenive operations) */ + * (e.g., injective, broadcast and elementwise without reduction, branch or expenive operations) + */ OperationMap is_strict_inlineable; /*! \brief Store whether the operation needs multi-level tiling * (e.g., computation-intensive ops with data reuse opportunity like matmul, conv2d) */ diff --git a/python/tvm/autotvm/task/relay_integration.py b/python/tvm/autotvm/task/relay_integration.py index 70f32eb81a75..9a43f2f1ad95 100644 --- a/python/tvm/autotvm/task/relay_integration.py +++ b/python/tvm/autotvm/task/relay_integration.py @@ -26,7 +26,6 @@ import tvm from .task import create from .topi_integration import TaskExtractEnv -from .dispatcher import FallbackContext logger = logging.getLogger('autotvm') diff --git a/src/auto_scheduler/compute_dag.cc b/src/auto_scheduler/compute_dag.cc index a7e0923285c8..68d1bb42c493 100644 --- a/src/auto_scheduler/compute_dag.cc +++ b/src/auto_scheduler/compute_dag.cc @@ -37,8 +37,8 @@ #include #include -#include "utils.h" #include "../arith/pattern_match.h" +#include "utils.h" namespace tvm { namespace auto_scheduler { @@ -517,7 +517,7 @@ bool AccessAnalyzer::ElementWiseMatch(const te::Operation& op, bool is_simple_access, axis_missing, axis_duplicated, same_order; for (const auto& read : reads) { is_simple_access = auto_scheduler::IsSimpleAccess(next_op, read, &axis_missing, - &axis_duplicated, &same_order); + &axis_duplicated, &same_order); if (!is_simple_access || axis_missing || axis_duplicated || !same_order) { return false; }