rogermt commited on
Commit
a284c9b
·
verified ·
1 Parent(s): 8051f1f

v5 refactor: add config.py

Browse files
Files changed (1) hide show
  1. neurogolf_solver/config.py +14 -0
neurogolf_solver/config.py ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """Runtime configuration management."""
3
+
4
+ from onnx import helper
5
+
6
+ def get_providers(device='auto'):
7
+ """Get ONNX Runtime execution providers based on device."""
8
+ if device == 'cuda':
9
+ return ['CUDAExecutionProvider', 'CPUExecutionProvider']
10
+ return ['CPUExecutionProvider']
11
+
12
+ def make_opset(version=17):
13
+ """Create ONNX opset identifier."""
14
+ return [helper.make_opsetid("", version)]