parkipy.utils.get_execution_space

parkipy.utils.get_execution_space(

execution_space: None | Literal['Cuda', 'HIP', 'OpenMP'] | pykokkos.ExecutionSpace = None,

) → pykokkos.ExecutionSpace

Return a Kokkos pk.ExecutionSpace object given am execution space specification.

Parameters:

execution_space (None | {"GPU", "Cuda", "HIP", "OpenMP"} | pk.ExecutionSpace) – Kokkos execution space specification. If None, return the Kokkos default execution space. The default is None. If “GPU”, choose one of “Cuda” or “HIP”, if available.

Return type:

pykokkos.ExecutionSpace