rogermt commited on
Commit
aa59e2e
·
verified ·
1 Parent(s): 2846f2b

Upload own-solver/neurogolf_solver/config.py

Browse files
own-solver/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)]