Exemplos de int_array em Python

Exemplo n.º 1

Arquivo: nccl.py Projeto: tate11/intelligent-code-completion

 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)))

Exemplo n.º 2

Arquivo: nccl.py Projeto: 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)))