ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #3664 from hoba87/develop

Browse Source 
simplify execution of mliap pytorch example
This commit is contained in:
Axel Kohlmeyer
2023-03-28 18:05:22 -04:00
committed by GitHub
parent 7d5d2b909f 88760170bf
commit f9edd3a539
3 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/lammps/mliap/__init__.py
View File

@ -7,18 +7,19 @@ import ctypes
import platform
import warnings
py_ver = sysconfig.get_config_vars('VERSION')[0]
py_ver = sysconfig.get_config_var('VERSION')
abi_flags = sysconfig.get_config_var('abiflags')
OS_name = platform.system()
if OS_name == "Darwin":
SHLIB_SUFFIX = '.dylib'
library = 'libpython' + py_ver + SHLIB_SUFFIX
library = 'libpython' + py_ver + abi_flags + SHLIB_SUFFIX
elif OS_name == "Windows":
SHLIB_SUFFIX = '.dll'
library = 'python' + py_ver + SHLIB_SUFFIX
library = 'python' + py_ver + abi_flags + SHLIB_SUFFIX
else:
SHLIB_SUFFIX = '.so'
library = 'libpython' + py_ver + SHLIB_SUFFIX
library = 'libpython' + py_ver + abi_flags + SHLIB_SUFFIX
try:
pylib = ctypes.CDLL(library)