Remove unnecessary import and fix linting

apache · Apr 17, 2024 · a94d539 · a94d539
1 parent 59b5b30
commit a94d539
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/python/tvm/topi/arm_cpu/conv2d.py b/python/tvm/topi/arm_cpu/conv2d.py
@@ -528,6 +528,7 @@ def compute_conv2d_NHWC(
     interleave_A,
     use_scalable_vectors=False,
 ):
+    """Compute definition for conv2d NHWC"""
     N, IH, IW, IC = get_const_tuple(data.shape)
     KH, KW, _, OC = get_const_tuple(kernel.shape)
     tile_N, tile_K = get_tiling_B_transformed(interleave_A, data.dtype, use_scalable_vectors)

diff --git a/src/arith/analyzer.cc b/src/arith/analyzer.cc
@@ -25,7 +25,6 @@
 #include <tvm/tir/expr.h>
 #include <tvm/tir/op.h>
 
-#include "../tir/analysis/check_contains.h"
 #include "./scalable_expression.h"
 #include "const_fold.h"
 #include "product_normal_form.h"
@@ -234,7 +233,7 @@ bool Analyzer::CanProve(const PrimExpr& expr, ProofStrength strength) {
   // "T.vscale" and the compile target uses a scalable architecture extension like
   // SVE, we can make some assumptions about the value of vscale and iterate over a
   // space of pre-defined values to attempt to prove the expression.
-  if (tir::CheckContains::ExprContains(simplified, IsVScaleCall)) {
+  if (ContainsVscaleCall(simplified)) {
     Target curr_target = tvm::Target::Current();
     if (curr_target.defined() && curr_target->features.defined() &&
         (curr_target->features.find("has_sve") != curr_target->features.end()) &&