Device Manager

Module: generative_models.core.device_manager

Source: generative_models/core/device_manager.py

Overview

Foundation-first device management system for Workshop.

This module provides a comprehensive, type-safe device management architecture that prioritizes clean design and robust error handling over backward compatibility.

Classes

CUDADetector

class CUDADetector

DeviceCapabilities

class DeviceCapabilities

DeviceConfiguration

class DeviceConfiguration

DeviceDetector

class DeviceDetector

DeviceManager

class DeviceManager

DeviceType

class DeviceType

JAXDeviceManager

class JAXDeviceManager

MemoryStrategy

class MemoryStrategy

Functions

init

def __init__()

init

def __init__()

configure_for_generative_models

def configure_for_generative_models()

cpu_devices

def cpu_devices()

detect_capabilities

def detect_capabilities()

detect_capabilities

def detect_capabilities()

device_count

def device_count()

devices

def devices()

distribute_data

def distribute_data()

get_default_device

def get_default_device()

get_default_device

def get_default_device()

get_device_info

def get_device_info()

get_device_manager

def get_device_manager()

gpu_count

def gpu_count()

gpu_devices

def gpu_devices()

has_gpu

def has_gpu()

has_gpu

def has_gpu()

optimize_for_model_size

def optimize_for_model_size()

print_device_info

def print_device_info()

Module Statistics

• Classes: 8
• Functions: 19
• Imports: 8