core.resharding

Subpackages

• core.resharding.nvshmem_copy_service
  • core.resharding.nvshmem_copy_service.core
    • core.resharding.nvshmem_copy_service.core.kernel_launcher
    • core.resharding.nvshmem_copy_service.core.gpu_resource_manager
    • core.resharding.nvshmem_copy_service.core.pipeline_executor
  • core.resharding.nvshmem_copy_service.planning
    • core.resharding.nvshmem_copy_service.planning.gpu_execution_planner
    • core.resharding.nvshmem_copy_service.planning.task_segmenter
    • core.resharding.nvshmem_copy_service.planning.communication_scheduler
    • core.resharding.nvshmem_copy_service.planning.workload_packer
  • core.resharding.nvshmem_copy_service.memory
    • core.resharding.nvshmem_copy_service.memory.double_buffer_manager
    • core.resharding.nvshmem_copy_service.memory.tensor_pointer_utils
  • core.resharding.nvshmem_copy_service.service
  • core.resharding.nvshmem_copy_service.nvshmem_types
  • core.resharding.nvshmem_copy_service.validation
  • core.resharding.nvshmem_copy_service.logger
• core.resharding.copy_services
  • core.resharding.copy_services.nccl_copy_service
  • core.resharding.copy_services.gloo_copy_service
  • core.resharding.copy_services.base
  • core.resharding.copy_services.nvshmem_copy_service

Submodules

• core.resharding.planner
• core.resharding.refit
• core.resharding.execution
• core.resharding.utils

Package Contents

Data

__all__

API

core.resharding.__all__

[‘build_centralized_reshard_plan’, ‘execute_reshard_plan’, ‘swap_model_weights’, ‘reshard_model_weig…