\[\newcommand{\vect}[1]{\vec{\boldsymbol{#1}}} \newcommand{\uvect}[1]{\hat{\boldsymbol{#1}}} \newcommand{\abs}[1]{\lvert#1\rvert} \newcommand{\Abs}[1]{\left\lvert#1\right\rvert} \newcommand{\norm}[1]{\lVert#1\rVert} \newcommand{\I}{\mathrm{i}} \newcommand{\ket}[1]{\left|#1\right\rangle} \newcommand{\bra}[1]{\left\langle#1\right|} \newcommand{\braket}[1]{\langle#1\rangle} \newcommand{\Braket}[1]{\left\langle#1\right\rangle} \newcommand{\op}[1]{\hat{#1}} \newcommand{\mat}[1]{\underline{\boldsymbol{#1}}} \providecommand{\d}{\mathrm{d}} \renewcommand{\d}{\mathrm{d}} \newcommand{\D}[1]{\mathcal{D}[#1]\;} \newcommand{\pdiff}[3][]{\frac{\partial^{#1}#2}{\partial{#3}^{#1}}} \newcommand{\diff}[3][]{\frac{\d^{#1} #2}{\d{#3}^{#1}}} \newcommand{\ddiff}[3][]{\frac{\delta^{#1} #2}{\delta{#3}^{#1}}} \newcommand{\floor}[1]{\left\lfloor#1\right\rfloor} \newcommand{\ceil}[1]{\left\lceil#1\right\rceil} \DeclareMathOperator{\Tr}{Tr} \DeclareMathOperator{\erf}{erf} \DeclareMathOperator{\erfi}{erfi} \DeclareMathOperator{\sech}{sech} \DeclareMathOperator{\sinc}{sinc} \DeclareMathOperator{\sn}{sn} \DeclareMathOperator{\cn}{cn} \DeclareMathOperator{\dn}{dn} \DeclareMathOperator{\sgn}{sgn} \DeclareMathOperator{\order}{O} \DeclareMathOperator{\diag}{diag} \DeclareMathOperator{\span}{span} \newcommand{\mylabel}[1]{\label{#1}\tag{#1}} \newcommand{\degree}{\circ} % These replace SIunitx. They should not be defined in LaTeX. \newcommand{\SI}[2]{#1\;\mathrm{#2}} \newcommand{\si}[1]{\mathrm{#1}} \let\qty\SI \let\unit\si \newcommand{\squared}{{^{2}}} \newcommand{\cubed}{{^{3}}} \newcommand{\per}{/} \newcommand{\tera}{T} \newcommand{\giga}{G} \newcommand{\mega}{M} \newcommand{\kilo}{k} \newcommand{\milli}{m} \newcommand{\micro}{μ} \newcommand{\nano}{n} \newcommand{\kilogram}{\text{kg}\,} \newcommand{\meter}{\text{m}\,} \newcommand{\second}{\text{s}\,} \newcommand{\ampere}{\text{A}\,} \newcommand{\kelvin}{\text{K}\,} \newcommand{\mol}{\text{mol}\,} \newcommand{\candela}{\text{cd}\,} \newcommand{\newton}{\text{N}\,} \newcommand{\hertz}{\text{Hz}\,} \newcommand{\pascal}{\text{Pa}\,} \newcommand{\volt}{\text{V}\,} \newcommand{\watt}{\text{W}\,} \newcommand{\joule}{\text{J}\,} \newcommand{\henry}{\text{H}\,} \newcommand{\farad}{\text{F}\,} \newcommand{\coulomb}{\text{C}\,} \newcommand{\ohm}{\Omega\,} \newcommand{\weber}{\text{Wb}\,} \newcommand{\tesla}{\text{T}\,} \newcommand{\degree}{\text{deg}\,} \]

gpe.gpu.bec

gpe.gpu.bec#

GPU versions of the code.

Requires cupy and the various NVIDIA dependencies.

Attributes#

u

Classes#

StateBase

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

Module Contents#

u[source]#

class StateBase(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)[source]#

Bases: pytimeode.mixins_gpu.GPUArrayStateMixin, gpe.bec.StateBase

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

xp[source]#

asnumpy[source]#

init()[source]#

Initialize the state.

This method defines the basis positions, momenta, etc. for use later on. We define these here rather than in the constructor __init__() so that the user can change them later and the reinitialize the state. We also call this function from the pre_evolve_hook() so that it is called before any evolution takes place. For this reason, we should not modify the state here.

get_metric_GPU()[source]#