an assert info change

Signed-off-by: Xiaowei Ren <xren@nvidia.com>
NVIDIA · Sep 25, 2024 · 1655edc · 1655edc
1 parent 1db4e8b
commit 1655edc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transformer_engine/pytorch/attention.py b/transformer_engine/pytorch/attention.py
@@ -4057,7 +4057,7 @@ def attn_forward_func_with_cp(
             cp_group = cp_group[0]
             cp_comm_type = "a2a"
     else:
-        assert isinstance(cp_group, dist_group_type), "Unsupported process group for non-hierarchical CP implementation!"
+        assert isinstance(cp_group, dist_group_type), f"Unsupported process group for CP communication type {cp_comm_type}!"
 
     assert qkv_format in [
         "bshd",