Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

if available, use MPI env vars to initialize multi-gpu configs #624

Closed
wants to merge 2 commits into from

Conversation

ngc92
Copy link
Contributor

@ngc92 ngc92 commented Jun 20, 2024

let's see what windows thinks of this

@ngc92 ngc92 changed the base branch from feature/nccl-only to master June 20, 2024 20:41
@ngc92 ngc92 changed the base branch from master to feature/nccl-only June 20, 2024 20:42
train_gpt2.cu Outdated
@@ -1334,6 +1334,12 @@ void error_usage() {
exit(EXIT_FAILURE);
}

int get_env_value(const char* env_name, int default_value) {
Copy link
Owner

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

windows looks happy. maybe get_env_int?

@ngc92 ngc92 closed this Jun 21, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants