diff --git a/python/sixtracklib/trackjob.py b/python/sixtracklib/trackjob.py index cd6f32f8..bef04dae 100644 --- a/python/sixtracklib/trackjob.py +++ b/python/sixtracklib/trackjob.py @@ -52,10 +52,6 @@ st_OutputBuffer_required_for_tracking, st_OutputBuffer_requires_output_buffer, st_OutputBuffer_calculate_output_buffer_params, - st_OpenCL_get_available_nodes_required_str_capacity, - st_OpenCL_get_all_nodes_required_str_capacity, - st_OpenCL_get_available_nodes_as_string, - st_OpenCL_get_all_nodes_as_string, st_TrackJob_get_type_str, st_TrackJobNew_track_elem_by_elem, st_TrackJobNew_track_line, @@ -142,6 +138,14 @@ st_TrackJobNew_uses_arguments, ) +if stconf.SIXTRACKLIB_MODULES.get("opencl", False): + from .stcommon import ( + st_OpenCL_get_available_nodes_required_str_capacity, + st_OpenCL_get_all_nodes_required_str_capacity, + st_OpenCL_get_available_nodes_as_string, + st_OpenCL_get_all_nodes_as_string, + ) + class TrackJobBaseNew(object): def __init__(self, ptr_track_job=None, owns_ptr=True):