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

make code python2/3 compatible

Browse Source
This commit is contained in:
Axel Kohlmeyer
2021-01-16 00:23:44 -05:00
parent efda7559c0
commit a644375afa
1 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
python/lammps/core.py
View File

@ -1449,16 +1449,16 @@ class lammps(object):
"""
result = {}
for p in [b'GPU', b'KOKKOS', b'USER-INTEL', b'USER-OMP']:
c = b'api'
result[p.decode()] = {}
for s in [b'cuda', b'hip', b'pthreads', b'opencl', b'openmp', b'serial']:
if self.lib.lammps_config_accelerator(p,c,s):
result[p.decode()][c.decode()] = s.decode()
c = b'precision'
for s in [b'double', b'mixed', b'single']:
if self.lib.lammps_config_accelerator(p,c,s):
result[p.decode()][c.decode()] = s.decode()
for p in ['GPU', 'KOKKOS', 'USER-INTEL', 'USER-OMP']:
c = 'api'
result[p] = {}
for s in ['cuda', 'hip', 'pthreads', 'opencl', 'openmp', 'serial']:
if self.lib.lammps_config_accelerator(p.encode(),c.encode(),s.encode()):
result[p][c] = s
c = 'precision'
for s in ['double', 'mixed', 'single']:
if self.lib.lammps_config_accelerator(p.encode(),c.encode(),s.encode()):
result[p][c] = s
return result
# -------------------------------------------------------------------------