Skip to content

Commit 5d30aa1

Browse files
refactor autosp config to permit user control over sp/dp and zero-1
1 parent 2459c00 commit 5d30aa1

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

benchmarks/autosp/configs/autosp_config.json

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,10 @@
88
"stage": 0
99
},
1010
"compile": {
11-
"deepcompile": true
11+
"deepcompile": true,
12+
"passes": ["autosp"],
13+
"sp_size": 2,
14+
"dp_size": 1
1215
},
1316
"gradient_accumulation_steps": 1,
1417
"gradient_clipping": "auto",

0 commit comments

Comments
 (0)