From 630affab17b860dc37e2deb6effef7e5337c8e18 Mon Sep 17 00:00:00 2001 From: Juhan Bae Date: Tue, 19 Mar 2024 03:55:18 -0400 Subject: [PATCH] Fix ddp argument bug --- tests/gpu_tests/ddp_variation_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/gpu_tests/ddp_variation_test.py b/tests/gpu_tests/ddp_variation_test.py index a27dacb..2b8e1dc 100644 --- a/tests/gpu_tests/ddp_variation_test.py +++ b/tests/gpu_tests/ddp_variation_test.py @@ -115,8 +115,8 @@ def test_lambda_matrices(self): for mp in module_partition_lst: for ca in cached_activation_cpu_offload: factor_args = FactorArguments( - covariance_data_partition_size=dp, - covariance_module_partition_size=mp, + lambda_data_partition_size=dp, + lambda_module_partition_size=mp, cached_activation_cpu_offload=ca, ) self.analyzer.fit_lambda_matrices(