deepmd_gnn.op

Contents

deepmd_gnn.op#

Load OP library.

Attributes#

SHARED_LIB_DIR

Functions#

load_library(→ None)

Load OP library.

Module Contents#

deepmd_gnn.op.SHARED_LIB_DIR[source]#

deepmd_gnn.op.load_library(module_name: str) → None[source]#

Load OP library.

Parameters#

module_namestr: Name of the module

Returns#

bool: Whether the library is loaded successfully