deepmd_gnn.op
=============

.. py:module:: deepmd_gnn.op

.. autoapi-nested-parse::

   Load OP library.



Attributes
----------

.. autoapisummary::

   deepmd_gnn.op.SHARED_LIB_DIR


Functions
---------

.. autoapisummary::

   deepmd_gnn.op.load_library


Module Contents
---------------

.. py:data:: SHARED_LIB_DIR

.. py:function:: load_library(module_name: str) -> None

   Load OP library.

   Parameters
   ----------
   module_name : str
       Name of the module

   Returns
   -------
   bool
       Whether the library is loaded successfully