remove dead code

andrej · andrej · commit efe8c619582e · 2026-03-11T16:47:55.000-06:00
diff --git a/iron/operators/axpy/design.py b/iron/operators/axpy/design.py
@@ -6,10 +6,8 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
-from aie.helpers.util import np_ndarray_type_get_shape
 
 
 def my_axpy(
diff --git a/iron/operators/dequant/design.py b/iron/operators/dequant/design.py
@@ -6,10 +6,8 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
-from aie.helpers.util import np_ndarray_type_get_shape
 
 
 def my_dequant_kernel(
diff --git a/iron/operators/elementwise_add/design.py b/iron/operators/elementwise_add/design.py
@@ -6,10 +6,8 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
-from aie.helpers.util import np_ndarray_type_get_shape
 
 
 def my_eltwise_add(
diff --git a/iron/operators/elementwise_mul/design.py b/iron/operators/elementwise_mul/design.py
@@ -6,10 +6,8 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
-from aie.helpers.util import np_ndarray_type_get_shape
 
 
 def my_eltwise_mul(
diff --git a/iron/operators/gelu/design.py b/iron/operators/gelu/design.py
@@ -7,7 +7,6 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import Tile, NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
 
diff --git a/iron/operators/gemv/design.py b/iron/operators/gemv/design.py
@@ -2,18 +2,12 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import numpy as np
-from pathlib import Path
 from ml_dtypes import bfloat16
-import argparse
 
 from aie.extras.context import mlir_mod_ctx
-from aie.ir import StridedLayoutAttr, ShapedType
-import aie.dialects.index as index
-import aie.dialects.memref as memref
 from aie.dialects.aie import *
 from aie.dialects.aiex import *
 from aie.helpers.dialects.scf import _for as range_
-from aie.helpers.util import try_convert_np_type_to_mlir_type
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
 from aie.iron.device import NPU1, NPU2
diff --git a/iron/operators/layer_norm/design.py b/iron/operators/layer_norm/design.py
@@ -6,10 +6,8 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
-from aie.helpers.util import np_ndarray_type_get_shape
 
 
 def my_layer_norm(
diff --git a/iron/operators/leaky_relu/design.py b/iron/operators/leaky_relu/design.py
@@ -6,7 +6,6 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import Tile, NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
 
diff --git a/iron/operators/mem_copy/design.py b/iron/operators/mem_copy/design.py
@@ -21,7 +21,6 @@
 from aie.iron.controlflow import range_
 from aie.iron.runtime.endpoint import RuntimeEndpoint
 from aie.iron.device import AnyShimTile
-import aie.dialects.index as index_dialect
 
 # The maximum value the 4th dimension of DMA BD can be set
 TAP_REPEAT_MAX = 64
diff --git a/iron/operators/mha/design.py b/iron/operators/mha/design.py
@@ -15,7 +15,6 @@
     Runtime,
     Worker,
     Buffer,
-    Buffer,
     WorkerRuntimeBarrier,
 )
 from aie.iron.placers import SequentialPlacer
diff --git a/iron/operators/relu/design.py b/iron/operators/relu/design.py
@@ -6,7 +6,6 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import Tile, NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
 
diff --git a/iron/operators/repeat/design.py b/iron/operators/repeat/design.py
@@ -3,10 +3,6 @@
 
 import numpy as np
 
-# from aie.extras.context import mlir_mod_ctx
-# from aie.ir import StridedLayoutAttr, ShapedType
-# from aie.dialects.aie import *
-# from aie.dialects.aiex import *
 from aie.dialects.aiex import TensorAccessPattern
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
diff --git a/iron/operators/rms_norm/design.py b/iron/operators/rms_norm/design.py
@@ -2,17 +2,13 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from ml_dtypes import bfloat16
-from pathlib import Path
 import numpy as np
-import argparse
-import sys
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
 from aie.iron.device import NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
-from aie.helpers.util import np_ndarray_type_get_shape
 
 
 def my_rms_norm(
diff --git a/iron/operators/rope/design.py b/iron/operators/rope/design.py
@@ -2,10 +2,7 @@
 # SPDX-License-Identifier: Apache-2.0
 
 
-from pathlib import Path
 import numpy as np
-import argparse
-import sys
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
diff --git a/iron/operators/sigmoid/design.py b/iron/operators/sigmoid/design.py
@@ -6,7 +6,6 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import Tile, NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
 
diff --git a/iron/operators/silu/design.py b/iron/operators/silu/design.py
@@ -6,10 +6,8 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import Tile, NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
-from aie.helpers.util import np_ndarray_type_get_shape
 
 
 def my_silu(
diff --git a/iron/operators/strided_copy/design.py b/iron/operators/strided_copy/design.py
@@ -3,10 +3,6 @@
 
 import numpy as np
 
-# from aie.extras.context import mlir_mod_ctx
-# from aie.ir import StridedLayoutAttr, ShapedType
-# from aie.dialects.aie import *
-# from aie.dialects.aiex import *
 from aie.dialects.aiex import TensorAccessPattern
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
diff --git a/iron/operators/tanh/design.py b/iron/operators/tanh/design.py
@@ -6,7 +6,6 @@
 
 from aie.iron import Kernel, ObjectFifo, Program, Runtime, Worker
 from aie.iron.placers import SequentialPlacer
-from aie.iron.device import Tile, NPU1, NPU2
 from aie.helpers.taplib.tap import TensorAccessPattern
 from aie.iron.controlflow import range_
 

Original file line number	Diff line number	Diff line change
`@@ -15,7 +15,6 @@`
`15`	`15`	`Runtime,`
`16`	`16`	`Worker,`
`17`	`17`	`Buffer,`
`18`		`- Buffer,`
`19`	`18`	`WorkerRuntimeBarrier,`
`20`	`19`	`)`
`21`	`20`	`from aie.iron.placers import SequentialPlacer`