From a94d539ae88cac7586614f77903a3334c3fac011 Mon Sep 17 00:00:00 2001 From: Andrei Hutu Date: Wed, 17 Apr 2024 14:14:04 +0000 Subject: [PATCH] Remove unnecessary import and fix linting --- python/tvm/topi/arm_cpu/conv2d.py | 1 + src/arith/analyzer.cc | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/topi/arm_cpu/conv2d.py b/python/tvm/topi/arm_cpu/conv2d.py index b6b5df8677ad5..44c4f7f76f693 100644 --- 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 index 84d987b199ef2..93409159e80f2 100644 --- a/src/arith/analyzer.cc +++ b/src/arith/analyzer.cc @@ -25,7 +25,6 @@ #include #include -#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()) &&