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

[#538]Add device_type function and warning messages #557

Merged

Conversation

Ideny42
Copy link
Contributor

@Ideny42 Ideny42 commented Jul 31, 2023

What changes were proposed in this pull request?

This pull request proposes a modification to the device_type function in the codebase. The changes involve determining the device type based on the input string, which is used to specify the device type (e.g., "gpu" , "Gpu01"or "cuda" for GPU, and "cpu" for CPU).

Why are the changes needed?

The purpose behind these changes is to improve the device type determination logic in the device_type function.

Does this PR introduce any user-facing change?

Yes, this PR introduces user-facing changes in the form of log messages. If the input string indicates a GPU device (e.g., "gpu" or "cuda"), the function now handles the variations in user input and provides a more informative message. If the input string is not recognized, the function also gives feedback about the unrecognized argument and defaults to using the CPU device.

How was this patch tested?

Use different parameter configurations for the 'device' argument and check the relevant logs.

Copy link
Collaborator

@Antlera Antlera left a comment

Choose a reason for hiding this comment

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

LGTM

@Antlera Antlera merged commit f6ea2db into intelligent-machine-learning:master Jul 31, 2023
13 checks passed
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.

2 participants