diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2020-11-26 11:42:04 +0100 |
---|---|---|
committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-12-07 14:51:52 +0000 |
commit | 1e17018d1aabff6b2a4cc5e8e3758678347b84c5 (patch) | |
tree | 8c06cb5a9f68e45fce96d9f17aac9a86f28ad912 /ethosu/vela/compiler_driver.py | |
parent | 32c7f5bbbccae480a0bb0c0e5b74a37dd9412023 (diff) | |
download | ethos-u-vela-1e17018d1aabff6b2a4cc5e8e3758678347b84c5.tar.gz |
MLBEDSW-3643: Refactor blockdep calculation
Moved blockdep calculation and other helper functions for
code generation to a separate file.
Change-Id: I2f8ccea478654272ebf42217fc5c1800e9ad177a
Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Diffstat (limited to 'ethosu/vela/compiler_driver.py')
-rw-r--r-- | ethosu/vela/compiler_driver.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index d17f1e5b..6c7fdc1a 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -20,6 +20,7 @@ import time from . import extract_npu_subgraphs from . import graph_optimiser from . import high_level_command_stream_generator +from . import high_level_command_to_npu_op from . import insert_dma from . import live_range from . import lut @@ -27,7 +28,6 @@ from . import mark_tensors from . import npu_performance from . import npu_serialisation from . import pass_packing -from . import register_command_stream_generator from . import scheduler from . import tensor_allocation from . import weight_compressor @@ -289,7 +289,7 @@ def compiler_driver(nng, arch, options, scheduler_options): nng, sg, arch, options.verbose_high_level_command_stream ) lut.optimize_high_level_cmd_stream(sg, arch) - register_command_stream_generator.generate_register_command_stream_for_sg( + high_level_command_to_npu_op.generate_register_command_stream_for_sg( nng, sg, arch, options.verbose_register_command_stream ) scratch_tens, scratch_fast_tens, flash_tens = npu_serialisation.serialise_npu_subgraph_into_tensors( |