diff options
author | Tim Hall <tim.hall@arm.com> | 2020-10-20 18:54:20 +0100 |
---|---|---|
committer | Tim Hall <tim.hall@arm.com> | 2020-10-21 15:23:33 +0100 |
commit | 4ed38bce498e1b9a5ae917316323de444792521a (patch) | |
tree | f3721d7131eeafa14c33cf0339d579de99a3c66a /ethosu/vela/npu_performance.py | |
parent | 9358296a51b9186335304a53bd7ea5dfbe5322d8 (diff) | |
download | ethos-u-vela-4ed38bce498e1b9a5ae917316323de444792521a.tar.gz |
vela: Refactor operators to use Kernel objects
- Normalise kernel availability by requiring all operators offer a kernel
describing how much data they consume from the source, per OFM element,
regardless of whether kernels are relevant to the operation.
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: Idbcff64879fc2eccf292b6208a7d2038eb388017
Diffstat (limited to 'ethosu/vela/npu_performance.py')
-rw-r--r-- | ethosu/vela/npu_performance.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py index e71e95b1..24b4c68a 100644 --- a/ethosu/vela/npu_performance.py +++ b/ethosu/vela/npu_performance.py @@ -31,7 +31,6 @@ from .nn_graph import PassPlacement from .nn_graph import SchedulerRewrite from .operation import NpuBlockType from .operation import Op -from .register_command_stream_generator import get_op_kernel from .tensor import MemArea from .tensor import shape_num_elements from .tensor import TensorBlockTraversal @@ -40,7 +39,7 @@ from .tensor import TensorPurpose def rolling_buffer_dims_from_passes(arch, ps1, block_config_ps1, ps2, block_config_ps2): ofm_block = Block(block_config_ps2[-3], block_config_ps2[-4], block_config_ps2[-1]) - kernel = get_op_kernel(ps2) + kernel = ps2.primary_op.kernel if ps2.npu_block_type in set((NpuBlockType.ConvolutionMxN, NpuBlockType.VectorProduct)): op = ps2.primary_op |