Skip to content

Consolidate logging for DLIOBenchmark initialization and data generation#8

Merged
FileSystemGuy merged 1 commit intomlcommons:mainfrom
wolfgang-desalvador:wdesalvador/consolidate-logging-on-rank-0
Apr 7, 2026
Merged

Consolidate logging for DLIOBenchmark initialization and data generation#8
FileSystemGuy merged 1 commit intomlcommons:mainfrom
wolfgang-desalvador:wdesalvador/consolidate-logging-on-rank-0

Conversation

@wolfgang-desalvador
Copy link
Copy Markdown

This pull request refactors how debug and warning messages are handled in the initialization and configuration of the DLIO benchmark. It replaces direct print statements with structured logging, ensures that debug output is only shown on the main process (rank 0), and clarifies the logic for selecting the data generation method.

Logging improvements:

  • Replaces all debug print statements in dlio_benchmark/main.py with calls to the DLIOLogger for consistent and centralized logging. Debug output is now only emitted by the main process (rank 0), reducing noise in multi-process runs. [1] [2]

Configuration logic and warnings:

  • In dlio_benchmark/utils/config.py, ensures that if the preferred data generation method is not available, it falls back to 'numpy' and logs a warning only from the main process. This prevents repeated warnings in multi-process scenarios and clarifies when and why the fallback occurs. [1] [2]…ion method

Copy link
Copy Markdown

@russfellows russfellows left a comment

Choose a reason for hiding this comment

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

I think these all look good. Agree we should be consistent with "self.logger" rather than print.

@wolfgang-desalvador wolfgang-desalvador force-pushed the wdesalvador/consolidate-logging-on-rank-0 branch from 3115892 to 6337267 Compare April 5, 2026 08:28
@FileSystemGuy FileSystemGuy merged commit 28b78ac into mlcommons:main Apr 7, 2026
3 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.

4 participants