예제 #1
0
 def __init__(self, devices):
     self.devices = devices
     ptrs = (NcclComm * len(devices))()
     self._as_parameter_ = ptrs
     check_error(lib.ncclCommInitAll(self, len(devices), int_array(devices)))
예제 #2
0
파일: nccl.py 프로젝트: athiwatp/pytorch
 def __init__(self, devices):
     self.devices = devices
     ptrs = (NcclComm * len(devices))()
     self._as_parameter_ = ptrs
     check_error(lib.ncclCommInitAll(self, len(devices), int_array(devices)))