diff options
Diffstat (limited to 'ethosu/vela/compiler_driver.py')
-rw-r--r-- | ethosu/vela/compiler_driver.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/compiler_driver.py b/ethosu/vela/compiler_driver.py index cec37ef8..1d8756b5 100644 --- a/ethosu/vela/compiler_driver.py +++ b/ethosu/vela/compiler_driver.py @@ -66,6 +66,7 @@ class CompilerOptions: timing=False, output_dir="outputs", cpu_tensor_alignment=Tensor.AllocationQuantum, + hillclimb_max_iterations=None, ): self.verbose_graph = verbose_graph @@ -84,6 +85,7 @@ class CompilerOptions: self.timing = timing self.output_dir = output_dir self.cpu_tensor_alignment = cpu_tensor_alignment + self.hillclimb_max_iterations = hillclimb_max_iterations def __str__(self): return type(self).__name__ + ": " + str(self.__dict__) |