fix config files

AlexMontgomerie · Mar 5, 2024 · 9e53340 · 9e53340
1 parent 4f75bdf
commit 9e53340
Show file tree

Hide file tree

Showing 6 changed files with 19 additions and 9 deletions.
diff --git a/examples/optimisers/greedy_partition_throughput.toml b/examples/optimisers/greedy_partition_throughput.toml
@@ -33,6 +33,8 @@ allowed_partitions = [
     ["EltWise", "*"]
 ]
 vertical = false
+allowed_single_layer_merge_pre = []
+allowed_single_layer_merge_post = []
 
 [annealing]
 T = 10.0

diff --git a/examples/optimisers/greedy_partition_throughput_efficientnet.toml b/examples/optimisers/greedy_partition_throughput_efficientnet.toml
@@ -30,6 +30,8 @@ allowed_partitions = [
     ["*", "Gemm"]
 ]
 vertical = false
+allowed_single_layer_merge_pre = []
+allowed_single_layer_merge_post = []
 
 [annealing]
 T = 10.0

diff --git a/examples/optimisers/greedy_partition_throughput_mobilenet.toml b/examples/optimisers/greedy_partition_throughput_mobilenet.toml
@@ -31,6 +31,8 @@ allowed_partitions = [
     ["*", "Gemm"]
 ]
 vertical = false
+allowed_single_layer_merge_pre = []
+allowed_single_layer_merge_post = []
 
 [annealing]
 T = 10.0

diff --git a/examples/optimisers/greedy_partition_throughput_resnet.toml b/examples/optimisers/greedy_partition_throughput_resnet.toml
@@ -30,6 +30,8 @@ allowed_partitions = [
     ["*", "Gemm"]
 ]
 vertical = false
+allowed_single_layer_merge_pre = []
+allowed_single_layer_merge_post = []
 
 [annealing]
 T = 10.0

diff --git a/examples/optimisers/greedy_partition_throughput_yolov3.toml b/examples/optimisers/greedy_partition_throughput_yolov3.toml
@@ -32,6 +32,8 @@ allowed_partitions = [
     ["Concat", "*"]
 ]
 vertical = true
+allowed_single_layer_merge_pre = []
+allowed_single_layer_merge_post = []
 
 [annealing]
 T = 10.0

diff --git a/examples/optimisers/single_partition_throughput.toml b/examples/optimisers/single_partition_throughput.toml
@@ -8,27 +8,27 @@ balance_bram_uram = true
 
 [transforms.fine]
 apply_transform = true
-probability = 0.25
+probability = 0.5
 start_complete = false
 
 [transforms.coarse]
 apply_transform = true
-probability = 0.25
+probability = 0.5
 
 [transforms.weights_reloading]
 apply_transform = false
-probability = 0.25
+probability = 0.0
 start_max = false
 
 [transforms.partition]
 apply_transform = false
-probability = 0.25
+probability = 0.0
 start_complete = false
-allowed_type = ["merge"]
-allowed_partitions = [
-    ["*", "Conv"],
-    ["*", "Gemm"]
-]
+allowed_type = []
+allowed_partitions = []
+vertical = false
+allowed_single_layer_merge_pre = []
+allowed_single_layer_merge_post = []
 
 [annealing]
 T = 10.0