def __init__(self, thread_count=None): if thread_count is None: thread_count = 0 if isinstance(thread_count, PythonWrapper.MathEngine): super().__init__(thread_count) return internal = PythonWrapper.MathEngine('cpu', int(thread_count), 0) super().__init__(internal)
def __init__(self, gpu_index): if gpu_index is None: gpu_index = 0 if isinstance(gpu_index, PythonWrapper.MathEngine): super().__init__(gpu_index) return if gpu_index < 0 or gpu_index >= len(enum_gpu()): raise ValueError("GPU with index `gpu_index` doesn't exist.") internal = PythonWrapper.MathEngine('gpu', 0, int(gpu_index)) super().__init__(internal)
def __init__(self, gpu_index=None): if gpu_index is None: gpu_index = -1 if isinstance(gpu_index, PythonWrapper.MathEngine): super().__init__(gpu_index) return gpus = enum_gpu() if len(gpus) == 0: raise ValueError("No GPU available") if gpu_index < -1 or gpu_index >= len(gpus): raise ValueError("GPU with index `gpu_index` doesn't exist.") internal = PythonWrapper.MathEngine('gpu', 0, int(gpu_index)) super().__init__(internal)