Update exports.

megatron/core/tensor_parallel/__init__.py

@@ -42,6 +42,7 @@ __all__ = [
     "ColumnParallelLinear",
     "RowParallelLinear",
     "VocabParallelEmbedding",
+    "set_tensor_model_parallel_attributes",
     "set_defaults_if_not_set_tensor_model_parallel_attributes",
     "copy_tensor_model_parallel_attributes",
     "param_is_not_tensor_parallel_duplicate",
@@ -59,4 +60,6 @@ __all__ = [
     "model_parallel_cuda_manual_seed",
     # utils.py
     "split_tensor_along_last_dim",
+    "split_tensor_into_1d_equal_chunks",
+    "gather_split_1d_tensor",
 ]