Create ipu_config.json
Browse files- ipu_config.json +9 -0
ipu_config.json
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"enable_half_partials": true,
|
3 |
+
|
4 |
+
"inference_replication_factor": {"pod16": 1, "pod32": 2, "pod64": 4, "default": 1},
|
5 |
+
"executable_cache_dir": "./exe_cache",
|
6 |
+
|
7 |
+
"layers_per_ipu": [2, 4, 3, 3, 3, 3, 3, 3, 2, 4, 3, 3, 3, 3, 3, 3],
|
8 |
+
"matmul_proportion": [0.1, 0.05, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2, 0.05, 0.05, 0.2, 0.2, 0.2, 0.2, 0.2, 0.2]
|
9 |
+
}
|