Skip to content

enhance logs with tflops/mfu#151

Open
ziqi-wlb wants to merge 2 commits intoByteDance-Seed:mainfrom
ziqi-wlb:feat/enhance_log_for_tflop
Open

enhance logs with tflops/mfu#151
ziqi-wlb wants to merge 2 commits intoByteDance-Seed:mainfrom
ziqi-wlb:feat/enhance_log_for_tflop

Conversation

@ziqi-wlb
Copy link
Copy Markdown
Contributor

What does this PR do?

  1. Add VEOMNI_GPU_TYPE environment to specific the GPU type ( Optional, for some scenes identify the GPU type incorrectly).
  2. Show tflops(per_rank) and mfu in training logs( For performance check).

API and Usage Example

# set real GPU type
export VEOMNI_GPU_TYPE=h100
...
  • as shown below:
企业微信截图_5e3fc85d-76d3-4598-831b-2a561605e3f5

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.

1 participant