Searched refs:initId (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/sys/objects/device/impls/cupm/ |
| H A D | cupmdevice.cxx | 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()
[all …]
|
| /petsc/src/sys/objects/device/impls/ |
| H A D | impldevicebase.hpp | 205 …tscDeviceInitType, PetscBool> &initType, std::pair<PetscInt, PetscBool> &initId, std::pair<PetscBo… in PetscOptionDeviceBasic() argument
209 PetscCall(PetscOptionDeviceSelect(PetscOptionsObject, &initId.first, &initId.second)); in PetscOptionDeviceBasic()
215 …tscDeviceInitType, PetscBool> &initType, std::pair<PetscInt, PetscBool> &initId, std::pair<PetscBo… in PetscOptionDeviceAll() argument
232 PetscCall(PetscOptionDeviceBasic(PetscOptionsObject, initType, initId, initView)); in PetscOptionDeviceAll()
|