Lines Matching refs:initId
247 auto initId = std::make_pair(*defaultDeviceId, PETSC_FALSE); in initialize() local
256 PetscCall(base_type::PetscOptionDeviceAll(comm, initType, initId, initView)); in initialize()
259 initId.first = PETSC_CUPM_DEVICE_NONE; in initialize()
267 initId.first = -static_cast<decltype(initId.first)>(cerr); in initialize()
273 if ((initId.first > 0) || (initId.first == PETSC_DECIDE)) initId.first = PETSC_CUPM_DEVICE_NONE; in initialize()
278 if (initId.first == PETSC_DECIDE) { in initialize()
287 initId.first = (PetscInt)strtol(pytorch_rank, &endptr, 10); in initialize()
288 …PetscCheck(initId.first < ndev, PETSC_COMM_SELF, PETSC_ERR_LIB, "PyTorch environmental variable LO… in initialize()
293 initId.first = rank % ndev; in initialize()
295 } else initId.first = 0; in initialize()
302 PetscCall(PetscMPIIntCast(initId.first, &defaultDevice_)); in initialize()
304 *defaultDeviceId = initId.first; in initialize()