clean up

Lightning-AI · Dec 20, 2024 · 5ce0da0 · 5ce0da0
1 parent 6d1c65b
commit 5ce0da0
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/thunder/examine/__init__.py b/thunder/examine/__init__.py
@@ -238,8 +238,6 @@ def get_nvfuser_fusion_definition(trace: TraceCtx, name: str, warn_if_fusion_una
         return None
 
     for bsym in trace.bound_symbols:
-        if bsym.sym.is_fusion:
-            print(bsym.sym.name)
         if bsym.sym.is_fusion and bsym.sym.name == name:
             _, fusion_ctx, _ = bsym.gather_ctxs()
             if (fusion_definition := fusion_ctx.get(name, None)) is not None:

diff --git a/thunder/tests/test_core.py b/thunder/tests/test_core.py
@@ -331,6 +331,7 @@ def _init_group(self, group, params, grads):
                     params.append(p)
                     grads.append(p.grad)
 
+        @torch.no_grad
         def step(self):
             for group in self.param_groups:
                 params = []