deepmd.pd.utils.env

deepmd.pd.utils.env#

Attributes#

`GLOBAL_NP_FLOAT_PRECISION`
`SAMPLER_RECORD`
`NUM_WORKERS`
`LOCAL_RANK`
`DEVICE`
`JIT`
`CINN`
`CINN_ALLOW_DYNAMIC_SHAPE`
`CACHE_PER_SYS`
`ENERGY_BIAS_TRAINABLE`
`CUSTOM_OP_USE_JIT`
`PRECISION_DICT`
`GLOBAL_PD_FLOAT_PRECISION`
`GLOBAL_PD_ENER_FLOAT_PRECISION`
`RESERVED_PRECISION_DICT`
`DEFAULT_PRECISION`

Functions#

enable_prim(→ None)

Module Contents#

deepmd.pd.utils.env.GLOBAL_NP_FLOAT_PRECISION[source]#

deepmd.pd.utils.env.SAMPLER_RECORD[source]#

deepmd.pd.utils.env.NUM_WORKERS[source]#

deepmd.pd.utils.env.LOCAL_RANK[source]#

deepmd.pd.utils.env.DEVICE = 'cpu'[source]#

deepmd.pd.utils.env.JIT[source]#

deepmd.pd.utils.env.CINN[source]#

deepmd.pd.utils.env.CINN_ALLOW_DYNAMIC_SHAPE[source]#

deepmd.pd.utils.env.CACHE_PER_SYS = 5[source]#

deepmd.pd.utils.env.ENERGY_BIAS_TRAINABLE = True[source]#

deepmd.pd.utils.env.CUSTOM_OP_USE_JIT[source]#

deepmd.pd.utils.env.PRECISION_DICT[source]#

deepmd.pd.utils.env.GLOBAL_PD_FLOAT_PRECISION[source]#

deepmd.pd.utils.env.GLOBAL_PD_ENER_FLOAT_PRECISION[source]#

deepmd.pd.utils.env.RESERVED_PRECISION_DICT[source]#

deepmd.pd.utils.env.DEFAULT_PRECISION = 'float64'[source]#

deepmd.pd.utils.env.enable_prim(enable: bool = True) → None[source]#