diff --git a/train/comms/pt/commsTraceParser.py b/train/comms/pt/commsTraceParser.py index 02d2b753..728fa929 100644 --- a/train/comms/pt/commsTraceParser.py +++ b/train/comms/pt/commsTraceParser.py @@ -23,6 +23,7 @@ "Tensor(c10::Half)": "half", "Tensor(c10::BFloat16)": "bfloat16", "Tensor(unsigned char)": "char", + "Tensor(signed char)": "char", }