lint

apache · Feb 11, 2025 · 9f2bde5 · 9f2bde5
1 parent bf77a5b
commit 9f2bde5
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 8 deletions.
diff --git a/python/tvm/relay/op/contrib/clml.py b/python/tvm/relay/op/contrib/clml.py
@@ -161,13 +161,6 @@ def partition_for_clml(mod, params=None, **opts):
     if params:
         mod["main"] = bind_params_by_name(mod["main"], params)
 
-    pass_context = tvm.get_global_func("transform.GetCurrentPassContext")()
-    target_version = (
-        pass_context.config["relay.ext.clml.target_version"]
-        if "relay.ext.clml.target_version" in pass_context.config
-        else 3
-    )
-
     seq = tvm.transform.Sequential(
         [
             transform.InferType(),

diff --git a/tests/python/contrib/test_hexagon/test_relax_2d_buffer_allocation.py b/tests/python/contrib/test_hexagon/test_relax_2d_buffer_allocation.py
@@ -17,6 +17,7 @@
 """Relax hexagon 2d VTCM allocation test."""
 
 import numpy as np
+import pytest
 
 import tvm
 import tvm.contrib.hexagon
@@ -25,7 +26,6 @@
 from tvm.script import ir as I
 from tvm.script import relax as R
 from tvm.script import tir as T
-import pytest
 
 
 # pylint: disable=missing-docstring,no-self-argument,invalid-name