remove _legacy_copy_params from add_transform

Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
Lightning-AI · Aug 21, 2024 · 76b910a · 76b910a
1 parent 3548ba8
commit 76b910a
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 8 deletions.
diff --git a/thunder/core/transforms.py b/thunder/core/transforms.py
@@ -406,7 +406,6 @@ def add_transform(
     *,
     transform: Transform | list[Transform],
     disable_torch_autograd_support=False,
-    _legacy_copy_params=False,
 ) -> Callable:
     from thunder.common import CompileData
 
@@ -438,11 +437,6 @@ def add_transform(
         disable_torch_autograd=cd.disable_torch_autograd_support or disable_torch_autograd_support,
         **cd.compile_options,
     )
-    from thunder import ThunderModule
-
-    if _legacy_copy_params and isinstance(jfn, ThunderModule):
-        jfn._overrides_parameters = cfn._overrides_parameters
-        jfn._overrides_buffers = cfn._overrides_buffers
     return jfn
 
 

diff --git a/thunder/distributed/tensor_parallel/column_wise.py b/thunder/distributed/tensor_parallel/column_wise.py
@@ -284,7 +284,6 @@ def forward(self, tokens: torch.Tensor) -> torch.Tensor:
             chunked_param_name_to_layer_type=chunked_param_name_to_layer_type,
             process_group=process_group,
         ),
-        _legacy_copy_params=True,
     )
 
     return colwise_thunder_module
diff --git a/thunder/distributed/tensor_parallel/row_wise.py b/thunder/distributed/tensor_parallel/row_wise.py
@@ -291,7 +291,6 @@ def forward(self, tokens: torch.Tensor) -> torch.Tensor:
             chunked_param_name_to_layer_type=chunked_param_name_to_layer_type,
             process_group=process_group,
         ),
-        _legacy_copy_params=True,
     )
 
     return rowwise_thunder_module