Skip to content

Conversation

@gaoteng-git
Copy link
Contributor

@gaoteng-git gaoteng-git marked this pull request as ready for review June 29, 2021 06:55
@gaoteng-git gaoteng-git requested a review from guotuofeng June 29, 2021 06:55
self.blocks_per_sm_count[gpu_id] += 1
else:
# Workaround for negative value input.
logger.warning("blocks per SM {} is negative!".format(blocks_per_sm))
Copy link
Contributor

Choose a reason for hiding this comment

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

log the timestamp or name for easily troubleshooting

if "blocks per SM" in event.args:
self.blocks_per_sm_count[gpu_id] += 1
blocks_per_sm = event.args.get("blocks per SM")
if blocks_per_sm >= 0.0:
Copy link
Contributor

Choose a reason for hiding this comment

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

do we need exclude blocks_per_sm is zero?

@gaoteng-git gaoteng-git merged commit e828a2c into pytorch:plugin/0.2 Jun 29, 2021
@gdankel
Copy link
Contributor

gdankel commented Jun 29, 2021

Thanks for the workaround - looks like the fix should be simple.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants