diff --git a/examples/optimisers/greedy_partition_throughput.toml b/examples/optimisers/greedy_partition_throughput.toml index 4d1e2603..a529060c 100644 --- 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 index 1bbaf63f..e2f935be 100644 --- 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 index 5d4beee3..f4b73cc3 100644 --- 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 index da37c64e..9bc746a6 100644 --- 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 index 9b1d1090..5dc91b6d 100644 --- 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 index dd92da69..92ecf4aa 100644 --- 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