From d032bd4dd1bff4427e16633c5fdf0e8c67eaa4f2 Mon Sep 17 00:00:00 2001
From: YangFei1990 <feiw@nvidia.com>
Date: Wed, 5 Aug 2026 15:51:13 -0700
Subject: [PATCH] expose _BOOTSTRAPPED

Signed-off-by: YangFei1990 <feiw@nvidia.com>
---
 transformer_engine/pytorch/ep.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/transformer_engine/pytorch/ep.py b/transformer_engine/pytorch/ep.py
index 940812b6a4..42f605367a 100644
--- a/transformer_engine/pytorch/ep.py
+++ b/transformer_engine/pytorch/ep.py
@@ -21,6 +21,7 @@
 __all__ = [
     "EpBuffer",
     "ep_bootstrap",
+    "is_ep_bootstrapped",
     "ep_finalize",
     "ep_dispatch",
     "ep_combine",
@@ -169,6 +170,11 @@ def ep_bootstrap(
         _ATEXIT_REGISTERED = True
 
 
+def is_ep_bootstrapped() -> bool:
+    """Whether EP has been initialized in this process."""
+    return _BOOTSTRAPPED
+
+
 def ep_finalize() -> None:
     """Optional explicit EP teardown; idempotent.