rename according to feedback

Lightning-AI · Nov 1, 2024 · a6d2d07 · a6d2d07
1 parent 8d803ae
commit a6d2d07
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/thunder/core/profile.py b/thunder/core/profile.py
@@ -52,11 +52,11 @@ def add_markers(msg: str) -> None:
 # but alternatively as a `with` context:
 #   with thunder.core.profile.profile("name for a block of code"):
 #      # ... code ...
-profile: Callable[[str], None] = None
+annotate_for_profile: Callable[[str], None] = None
 
 
 if _ENABLED:
-    profile = functools.partial(nvtx.annotate, domain="thunder")
+    annotate_for_profile = functools.partial(nvtx.annotate, domain="thunder")
 else:
 
     class _no_annotate(contextlib.nullcontext):
@@ -70,4 +70,4 @@ def __init__(self, *args, **kwargs):
         def __call__(self, fqn):
             return fqn
 
-    profile = _no_annotate
+    annotate_for_profile = _no_annotate
diff --git a/thunder/tests/test_core.py b/thunder/tests/test_core.py
@@ -2970,7 +2970,7 @@ def fn():
 
 
 def test_profiling_decorator():
-    @thunder.core.profile.profile("compile_and_run")
+    @thunder.core.profile.annotate_for_profile("compile_and_run")
     def foo():
         def bar(a: torch.Tensor):
             t0 = torch.add(a, 42)