Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.
GNU General Public License v3.0
35
stars
14
forks
source link
Fix vectorized tracking for `TransverseDeflectingCavity` #278
Closed
cr-xu closed 1 month ago
Description
Adds some missing dimensional expansions in the
TransverseDeflectingCavity
code to fix a bug with some vectorisation setups.Motivation and Context
Fixes #270.
Types of changes
Checklist
flake8
(required).pytest
tests pass (required).pytest
on a machine with a CUDA GPU and made sure all tests pass (required).Note: We are using a maximum length of 88 characters per line.