From efdc623d2b15ae1b7610b47436c0843b0fd58a61 Mon Sep 17 00:00:00 2001 From: Can Zhao <69829124+Can-Zhao@users.noreply.github.com> Date: Thu, 6 Feb 2025 12:03:20 -0800 Subject: [PATCH 1/5] Update autoencoderkl_maisi.py When there is no need to perform tensor parallel, skip all the unnecessary splitting steps --- monai/apps/generation/maisi/networks/autoencoderkl_maisi.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py index 6251ea8e83..3ed9c03188 100644 --- a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py +++ b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py @@ -231,6 +231,10 @@ def _concatenate_tensors(self, outputs: list[torch.Tensor], split_size: int, pad def forward(self, x: torch.Tensor) -> torch.Tensor: if self.print_info: logger.info(f"Number of splits: {self.num_splits}") + + if self.dim_split<=1 and self.num_splits<=1: + x = self.conv(x) + return x # compute size of splits l = x.size(self.dim_split + 2) From 866f1942140a556cf8ab1308cca5c3396fda2c1a Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Thu, 6 Feb 2025 20:05:03 +0000 Subject: [PATCH 2/5] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- monai/apps/generation/maisi/networks/autoencoderkl_maisi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py index 3ed9c03188..141f77231e 100644 --- a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py +++ b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py @@ -231,7 +231,7 @@ def _concatenate_tensors(self, outputs: list[torch.Tensor], split_size: int, pad def forward(self, x: torch.Tensor) -> torch.Tensor: if self.print_info: logger.info(f"Number of splits: {self.num_splits}") - + if self.dim_split<=1 and self.num_splits<=1: x = self.conv(x) return x From 131f7e650fb85b0cf36f5eb8509ec3b242495357 Mon Sep 17 00:00:00 2001 From: Can Zhao <69829124+Can-Zhao@users.noreply.github.com> Date: Thu, 6 Feb 2025 22:34:25 -0800 Subject: [PATCH 3/5] Update autoencoderkl_maisi.py --- monai/apps/generation/maisi/networks/autoencoderkl_maisi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py index 141f77231e..86b4e68864 100644 --- a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py +++ b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py @@ -232,7 +232,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: if self.print_info: logger.info(f"Number of splits: {self.num_splits}") - if self.dim_split<=1 and self.num_splits<=1: + if self.dim_split <= 1 and self.num_splits <= 1: x = self.conv(x) return x From 06d8661b90f9465f5e42c40e0ecff13d0821aba4 Mon Sep 17 00:00:00 2001 From: Can Zhao <69829124+Can-Zhao@users.noreply.github.com> Date: Mon, 10 Feb 2025 20:29:50 -0800 Subject: [PATCH 4/5] DCO Remediation Commit for Can Zhao <69829124+Can-Zhao@users.noreply.github.com> I, Can Zhao <69829124+Can-Zhao@users.noreply.github.com>, hereby add my Signed-off-by to this commit: efdc623d2b15ae1b7610b47436c0843b0fd58a61 I, Can Zhao <69829124+Can-Zhao@users.noreply.github.com>, hereby add my Signed-off-by to this commit: 131f7e650fb85b0cf36f5eb8509ec3b242495357 Signed-off-by: Can Zhao <69829124+Can-Zhao@users.noreply.github.com> --- monai/apps/generation/maisi/networks/autoencoderkl_maisi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py index 86b4e68864..aa25adc50b 100644 --- a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py +++ b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py @@ -234,7 +234,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: if self.dim_split <= 1 and self.num_splits <= 1: x = self.conv(x) - return x + return x # compute size of splits l = x.size(self.dim_split + 2) From 06321d32000663620c26d9ea494a5290cf68dcac Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Tue, 11 Feb 2025 04:30:12 +0000 Subject: [PATCH 5/5] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- monai/apps/generation/maisi/networks/autoencoderkl_maisi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py index aa25adc50b..86b4e68864 100644 --- a/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py +++ b/monai/apps/generation/maisi/networks/autoencoderkl_maisi.py @@ -234,7 +234,7 @@ def forward(self, x: torch.Tensor) -> torch.Tensor: if self.dim_split <= 1 and self.num_splits <= 1: x = self.conv(x) - return x + return x # compute size of splits l = x.size(self.dim_split + 2)