From 47dcb8ebcd613abfc2649fa657729e759f449f66 Mon Sep 17 00:00:00 2001
From: Chaitanya Sri Krishna Lolla <lollachaitanya@gmail.com>
Date: Wed, 10 Apr 2024 14:47:55 +0530
Subject: [PATCH] Update bfloat16 type

---
 rccl_nccl_parser.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/rccl_nccl_parser.py b/rccl_nccl_parser.py
index bd11836..a4c0823 100644
--- a/rccl_nccl_parser.py
+++ b/rccl_nccl_parser.py
@@ -34,7 +34,7 @@
                 "6" : "half",
                 "7" : "float",
                 "8" : "double",
-                "9" : "bf16",
+                "9" : "bfloat16",
                 #"10" : "ncclNumTypes Equivalent?"
              }