Skip to content

Fix missing device in FixCom and get_centers_of_mass()#583

Open
sklenard wants to merge 1 commit into
TorchSim:mainfrom
sklenard:bugfix/fixcom-device-mismatch
Open

Fix missing device in FixCom and get_centers_of_mass()#583
sklenard wants to merge 1 commit into
TorchSim:mainfrom
sklenard:bugfix/fixcom-device-mismatch

Conversation

@sklenard

@sklenard sklenard commented Jul 1, 2026

Copy link
Copy Markdown

Summary

Related to #580.
Add device to all torch.zeros() calls in FixCom and in get_center_of_mass, fixing a crash on CUDA when using the FixCom constraint.

Checklist

Before a pull request can be merged, the following items must be checked:

  • Doc strings have been added in the Google docstring format.
  • Run ruff on your code.
  • Tests have been added for any new functionality or bug fixes.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant