TorchAutogradFunction"X"txProxyInterface -> TorchAutogradFunction"C"t…

…xProxyInterface (#1147)
Lightning-AI · Sep 14, 2024 · 0073731 · 0073731
1 parent 177643b
commit 0073731
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/thunder/core/baseutils.py b/thunder/core/baseutils.py
@@ -75,7 +75,7 @@ class TensorProxyInterface:
     pass
 
 
-class TorchAutogradFunctionXtxProxyInterface:
+class TorchAutogradFunctionCtxProxyInterface:
     pass
 
 

diff --git a/thunder/core/proxies.py b/thunder/core/proxies.py
@@ -20,7 +20,7 @@
     ProxyInterface,
     NumberProxyInterface,
     TensorProxyInterface,
-    TorchAutogradFunctionXtxProxyInterface,
+    TorchAutogradFunctionCtxProxyInterface,
 )
 import thunder.core.baseutils as baseutils
 from thunder.core.langctxs import resolve_method, get_langctx
@@ -1858,7 +1858,7 @@ def real(self):
         return method(self)
 
 
-class TorchAutogradFunctionCtxProxy(Proxy, TorchAutogradFunctionXtxProxyInterface):
+class TorchAutogradFunctionCtxProxy(Proxy, TorchAutogradFunctionCtxProxyInterface):
     def __init__(
         self,
         ctx: torch.autograd.function.FunctionCtx,