gpe.gpu.bec2

GPU versions of the code.

Requires cupy and the various NVIDIA dependencies.

Attributes

u

Classes

State

Mixin providing support for states with a single GPU data array.

Module Contents

u

class State(basis=None, Nxyz=(2**5, 2**5, 2**5), Lxyz=(30 * u.micron, 50 * u.micron, 50 * u.micron), symmetric_grid=False, twist=None, **kw)

Bases: pytimeode.mixins_gpu.GPUArrayStateMixin, gpe.bec2.State

Mixin providing support for states with a single GPU data array.

xp

asnumpy