core.resharding

Subpackages

• core.resharding.nvshmem_copy_service
  • 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.core
    • core.resharding.nvshmem_copy_service.core.kernel_launcher
    • core.resharding.nvshmem_copy_service.core.pipeline_executor
    • core.resharding.nvshmem_copy_service.core.gpu_resource_manager
  • core.resharding.nvshmem_copy_service.planning
    • core.resharding.nvshmem_copy_service.planning.workload_packer
    • 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.compat
  • core.resharding.nvshmem_copy_service.validation
  • core.resharding.nvshmem_copy_service.logger
  • core.resharding.nvshmem_copy_service.nvshmem_types
  • core.resharding.nvshmem_copy_service.service
• core.resharding.copy_services
  • core.resharding.copy_services.nixl_copy_service
  • core.resharding.copy_services.gloo_copy_service
  • core.resharding.copy_services.nvshmem_copy_service
  • core.resharding.copy_services.nccl_copy_service
  • core.resharding.copy_services.base

Submodules

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

Package Contents

Data

__all__

API

core.resharding.__all__

[‘build_centralized_reshard_plan’, ‘build_local_reshard_plan’, ‘build_plan_from_rosters’, ‘index_met…