Merge pull request #10 from ROCm/cl/update-dtype

Update str to nccl bfloat16 dtype for parsing RCCL/NCCL configurations.
ROCm · Apr 10, 2024 · ceb828b · ceb828b
2 parents e632f8d + 47dcb8e
commit ceb828b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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?"
              }