Skip to content

Conversation

@gaoteng-git
Copy link
Contributor

@gaoteng-git gaoteng-git commented May 11, 2021

Device properties are useful in any case, but especially for performing analysis on traces such as occupancy.

This patch is a re-implementation of #209
And it added 2 fields "name" and "totalGlobalMem", to remove duplicate from PyTorch side dump.
It adds "ifdef" to ignore dumping these fields when build on pure cpu.

Summary:
Device properties are useful in any case, but especially for performing analysis on traces such as occupancy.

This patch is a re-implementation of pytorch#209

Reviewed By: ilia-cher

Differential Revision: D28337067

fbshipit-source-id: b61588b414a9faa6f697260b48f750f013ba553b
@gaoteng-git gaoteng-git marked this pull request as ready for review May 11, 2021 11:15
@gaoteng-git gaoteng-git requested a review from gdankel May 11, 2021 11:15

#ifdef HAS_CUPTI
traceOf_ << fmt::format(R"JSON(
"computeProperties": [
Copy link
Contributor

Choose a reason for hiding this comment

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

need confirm with @gdankel about using top level property or putting it into metadata.

My personal preference is put it into metadata.

@guotuofeng guotuofeng closed this May 12, 2021
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