Skip to content

Commit

Permalink
Add Tensor(signed char)
Browse files Browse the repository at this point in the history
Summary: The most popular trace (according to this dashboard: https://fburl.com/unidash/iv4evgh6) is currently unsupported by param trace because we get a key error when encountering the `Tensor(signed char)` type. This diff resolves this issue.

Reviewed By: bahlneeraj

Differential Revision: D56451067

fbshipit-source-id: a9ef08d7b73b665d7f6e2f4414aec961b9548589
  • Loading branch information
Ahmed Khan authored and facebook-github-bot committed Apr 23, 2024
1 parent 05abe4e commit 6c0067f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions train/comms/pt/commsTraceParser.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@
"Tensor(c10::Half)": "half",
"Tensor(c10::BFloat16)": "bfloat16",
"Tensor(unsigned char)": "char",
"Tensor(signed char)": "char",
}


Expand Down

0 comments on commit 6c0067f

Please sign in to comment.