Description

Please summarise the changes.

Related issues

Please mention any github issues addressed by this PR.

Checklist

I have run the tests on a device with GPUs.
I have performed a self-review of my code.
I have commented hard-to-understand parts of my code.
I have made corresponding changes to the public API documentation.
I have added tests that prove my fix is effective or that my feature works.
I have updated the changelog with any user-facing changes.