Searched refs:ndev (Results 1 – 1 of 1) sorted by relevance
250 int ndev = 0; in initialize() local260 } else if (const auto cerr = cupmGetDeviceCount(&ndev)) { in initialize()277 PetscCall(PetscDeviceCheckDeviceCount_Internal(ndev)); in initialize()279 if (ndev) { in initialize()288 …nitId.first < ndev, PETSC_COMM_SELF, PETSC_ERR_LIB, "PyTorch environmental variable LOCAL_RANK %s … in initialize()293 initId.first = rank % ndev; in initialize()