AttributeError when using GPU PyHeadTail

[ustc-cheese]

Hello everyone~

When I am using the GPU version of PyHeadtail, an error has been encountered when using
with GPU(particles) as context:
for turn in range(n_turns):
machine.track(particles)
AttributeError: module 'PyHEADTAIL.general.pmath' has no attribute '_GPU_func_dict'
How can I resolve the problem? We will appreciate your help~ Thank you.

Best

Weijie