Fixed linting

Change-Id: I060a2f95fb31beedb9e6ac8c8b45d51d6ec7ebce

python/tvm/driver/build_module.py

@@ -183,6 +183,7 @@ def lower(sch, args, name="main", binds=None, simple_mode=False):
     pass_list += [
         tvm.tir.transform.VectorizeLoop(not disable_vectorize),
         tvm.tir.transform.InjectVirtualThread(),
+        tvm.tir.transform.InjectRollingBuffer(),
         tvm.tir.transform.InjectDoubleBuffer(),
         tvm.tir.transform.StorageRewrite(),
         tvm.tir.transform.UnrollLoop(),

python/tvm/tir/transform/inject_rolling_buffer.py

@@ -14,11 +14,13 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
+"""Inject rolling buffers through a TIR transformation."""
+# pylint: disable=invalid-name,unused-argument,inconsistent-return-statements
+from collections import defaultdict
+
 import tvm
 from tvm import arith
 
-from collections import defaultdict
-
 
 def InjectRollingBuffer():
     """Inject rolling buffer statements.
@@ -48,7 +50,7 @@ def _pre_visit(stmt):
 
         elif isinstance(stmt, tvm.tir.AttrStmt):
             if isinstance(stmt.node, tvm.tir.Buffer):
-                if stmt.attr_key == "rolling_buffer" and stmt.value.value == True:
+                if stmt.attr_key == "rolling_buffer" and stmt.value.value:
                     # If the attribute is indicating that a buffer should be a rolling
                     # buffer, then update the rolling_buffers set to include the bufffer
                     rolling_buffers.add(stmt.node)

python/tvm/tir/transform/transform.py

@@ -15,7 +15,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """Wrapping existing transformations."""
-# pylint: disable=invalid-name
+# pylint: disable=invalid-name,unused-import
 from . import _ffi_api
 from . import function_pass as _fpass
 from .inject_rolling_buffer import InjectRollingBuffer