Browse Source

feat(mge/distributed): deprecate get_device_count_by_fork

GitOrigin-RevId: 6147c3ae90
release-1.5
Megvii Engine Team 3 years ago
parent
commit
b17a02d44a
1 changed files with 6 additions and 1 deletions
  1. +6
    -1
      imperative/python/megengine/distributed/helper.py

+ 6
- 1
imperative/python/megengine/distributed/helper.py View File

@@ -15,12 +15,12 @@ from weakref import WeakSet
import numpy as np import numpy as np


from megengine.autodiff.grad_manager import GradManager, get_backwarding_grad_manager from megengine.autodiff.grad_manager import GradManager, get_backwarding_grad_manager
from megengine.device import get_default_device, get_device_count


from ..core._imperative_rt.core2 import apply from ..core._imperative_rt.core2 import apply
from ..core.ops.builtin import ParamPackConcat, ParamPackSplit from ..core.ops.builtin import ParamPackConcat, ParamPackSplit
from ..functional.tensor import copy from ..functional.tensor import copy
from ..tensor import Tensor from ..tensor import Tensor
from ..utils.deprecation import deprecated_func
from ..utils.future import Future from ..utils.future import Future
from . import group as _group from . import group as _group
from .functional import _bcast_param, all_reduce_sum, broadcast from .functional import _bcast_param, all_reduce_sum, broadcast
@@ -193,6 +193,11 @@ def _check_device_initialized(device_type: str, rank: int):
raise RuntimeError(errmsg) raise RuntimeError(errmsg)




get_device_count_by_fork = deprecated_func(
"1.5", "megengine.device", "get_device_count", False
)


def bcast_list_(inps: list, group: Group = WORLD): def bcast_list_(inps: list, group: Group = WORLD):
""" """
Broadcast tensors between given group. Broadcast tensors between given group.


Loading…
Cancel
Save