Searched refs:pytorch_rank (Results 1 – 1 of 1) sorted by relevance
282 char *pytorch_rank = (char *)getenv("LOCAL_RANK"); in initialize() local284 if (pytorch_exists && pytorch_rank) { in initialize()287 initId.first = (PetscInt)strtol(pytorch_rank, &endptr, 10); in initialize()288 …C_ERR_LIB, "PyTorch environmental variable LOCAL_RANK %s > number devices %d", pytorch_rank, ndev); in initialize()