fix(transforms): preserve device and dtype in squarepulse

[aymuos15]

Description

squarepulse allocated its output via torch.zeros(t.shape), which defaults to a CPU float32 tensor regardless of the input tensor's device or dtype. This caused CUDA callers to silently receive a CPU result, breaking downstream transforms when GPU tensors were passed.

This fix allocates the output on the input tensor's device, and preserves the input's floating-point dtype. Integer inputs still promote to the default float dtype (matching torch.zeros behavior).

Diff

-    y = torch.zeros(t.shape)
+    y = torch.zeros(t.shape, device=t.device, dtype=t.dtype if t.is_floating_point() else None)

Types of changes

• Non-breaking change (fix or new feature that would not break existing functionality).
• Breaking change (fix or new feature that would cause existing functionality to change).
• New tests added to cover the changes.
• Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
• Quick tests passed locally by running ./runtests.sh --quick --unittests --disttests.
• In-line docstrings updated.
• Documentation updated, tested make html command in the docs/ folder.

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro Plus

Run ID: 50de0856-146c-42ae-ac27-2cf545cbf71d

📥 Commits

Reviewing files that changed from the base of the PR and between 2529047 and ae0021c.

📒 Files selected for processing (1)

• monai/transforms/utils.py

🚧 Files skipped from review as they are similar to previous changes (1)

• monai/transforms/utils.py

Included review availability: Your plan includes up to 8 reviews per rolling hour; 5 remain after this review.

---

📝 Walkthrough

Walkthrough

Squarepulse now creates its output tensor on the input device and preserves floating-point input dtype. Integer inputs retain default dtype promotion. New tests cover CPU and CUDA inputs, output shape, dtype, device placement, binary values, and reference-value matching.

Estimated code review effort: 2 (Simple) | ~10 minutes

Merge Risk: ⚪ Minimal · up to ae002

The change preserves the input tensor's device and floating-point dtype while retaining existing integer promotion behavior; no actionable merge-blocking risk remains after normal checks and review.

🚥 Pre-merge checks | ✅ 4 | ❌ 1

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Docstring Coverage	⚠️ Warning	Docstring coverage is 33.33% which is insufficient. The required threshold is 80.00%.	Write docstrings for the functions missing them to satisfy the coverage threshold.

✅ Passed checks (4 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title clearly and concisely describes preserving device and dtype in squarepulse, which is the main change.
Description check	✅ Passed	The description explains the bug and fix, includes the relevant diff, identifies the change type, and records new tests.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands.}