fix: correct MPS execution on apple silicon#145

Open

yanghan234 wants to merge 7 commits intomainfrom

hanyang/fix-mps-device-handling

Commits on Apr 7, 2026

fix: correct MPS execution in m3gnet forcefield
yanghan234
committed
fix: prevent device mismatch in batch_to_dict and get_properties on CUDA

yanghan234
and
Copilot
committed
refactor: fold move_to_device into batch_to_dict

yanghan234
and
Copilot
committed
test: parametrize tests by available device (cpu/mps/cuda)

yanghan234
and
Copilot
committed
test: log device name during test execution

yanghan234
and
Copilot
committed
fix: add backward-compat fallbacks and clean up review findings

yanghan234
and
Copilot
committed
fix: create index_map on same device as num_triple_ij in batch_to_dict

yanghan234
and
Copilot
committed