Select the appropriate shared library suffix based on the operating system

hlyang1992 · Han Wang · commit 7a31fdb071f7 · 2019-10-11T18:49:38.000+08:00
diff --git a/source/op/_prod_force_grad.py b/source/op/_prod_force_grad.py
@@ -4,13 +4,20 @@
 """
 
 import os
+import platform
 import tensorflow as tf
 from tensorflow.python.framework import ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import sparse_ops
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__))
-module_file = os.path.join(module_path, 'libop_grads.so')
+module_file = os.path.join(module_path, 'libop_grads.{}'.format(ext))
 assert (os.path.isfile(module_file)), 'module op_grads does not exist'
 op_grads_module = tf.load_op_library(module_file)
      
diff --git a/source/op/_prod_force_se_a_grad.py b/source/op/_prod_force_se_a_grad.py
@@ -4,13 +4,20 @@
 """
 
 import os
+import platform
 import tensorflow as tf
 from tensorflow.python.framework import ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import sparse_ops
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__))
-module_file = os.path.join(module_path, 'libop_grads.so')
+module_file = os.path.join(module_path, 'libop_grads.{}'.format(ext))
 assert (os.path.isfile(module_file)), 'module op_grads does not exist'
 op_grads_module = tf.load_op_library(module_file)
      
diff --git a/source/op/_prod_force_se_r_grad.py b/source/op/_prod_force_se_r_grad.py
@@ -4,13 +4,20 @@
 """
 
 import os
+import platform
 import tensorflow as tf
 from tensorflow.python.framework import ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import sparse_ops
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__))
-module_file = os.path.join(module_path, 'libop_grads.so')
+module_file = os.path.join(module_path, 'libop_grads.{}'.format(ext))
 assert (os.path.isfile(module_file)), 'module op_grads does not exist'
 op_grads_module = tf.load_op_library(module_file)
      
diff --git a/source/op/_prod_virial_grad.py b/source/op/_prod_virial_grad.py
@@ -4,13 +4,20 @@
 """
 
 import os
+import platform
 import tensorflow as tf
 from tensorflow.python.framework import ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import sparse_ops
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__))
-module_file = os.path.join(module_path, 'libop_grads.so')
+module_file = os.path.join(module_path, 'libop_grads.{}'.format(ext))
 assert (os.path.isfile(module_file)), 'module op_grads does not exist'
 op_grads_module = tf.load_op_library(module_file)
      
diff --git a/source/op/_prod_virial_se_a_grad.py b/source/op/_prod_virial_se_a_grad.py
@@ -4,13 +4,20 @@
 """
 
 import os
+import platform
 import tensorflow as tf
 from tensorflow.python.framework import ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import sparse_ops
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__))
-module_file = os.path.join(module_path, 'libop_grads.so')
+module_file = os.path.join(module_path, 'libop_grads.{}'.format(ext))
 assert (os.path.isfile(module_file)), 'module op_grads does not exist'
 op_grads_module = tf.load_op_library(module_file)
      
diff --git a/source/op/_prod_virial_se_r_grad.py b/source/op/_prod_virial_se_r_grad.py
@@ -4,13 +4,20 @@
 """
 
 import os
+import platform
 import tensorflow as tf
 from tensorflow.python.framework import ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import sparse_ops
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__))
-module_file = os.path.join(module_path, 'libop_grads.so')
+module_file = os.path.join(module_path, 'libop_grads.{}'.format(ext))
 assert (os.path.isfile(module_file)), 'module op_grads does not exist'
 op_grads_module = tf.load_op_library(module_file)
      
diff --git a/source/op/_soft_min_force_grad.py b/source/op/_soft_min_force_grad.py
@@ -4,13 +4,20 @@
 """
 
 import os
+import platform
 import tensorflow as tf
 from tensorflow.python.framework import ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import sparse_ops
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__))
-module_file = os.path.join(module_path, 'libop_grads.so')
+module_file = os.path.join(module_path, 'libop_grads.{}'.format(ext))
 assert (os.path.isfile(module_file)), 'module op_grads does not exist'
 op_grads_module = tf.load_op_library(module_file)
      
diff --git a/source/op/_soft_min_virial_grad.py b/source/op/_soft_min_virial_grad.py
@@ -4,13 +4,20 @@
 """
 
 import os
+import platform
 import tensorflow as tf
 from tensorflow.python.framework import ops
 from tensorflow.python.ops import array_ops
 from tensorflow.python.ops import sparse_ops
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__))
-module_file = os.path.join(module_path, 'libop_grads.so')
+module_file = os.path.join(module_path, 'libop_grads.{}'.format(ext))
 assert (os.path.isfile(module_file)), 'module op_grads does not exist'
 op_grads_module = tf.load_op_library(module_file)
      
diff --git a/source/scripts/freeze.py b/source/scripts/freeze.py
@@ -3,6 +3,7 @@
 # freeze.py :
 # see https://blog.metaflow.fr/tensorflow-how-to-freeze-a-model-and-serve-it-with-a-python-api-d4f3596b3adc
 
+import platform
 import os, argparse, json
 import sys
 
@@ -13,9 +14,15 @@
 from tensorflow.python.framework import ops
 
 # load force module
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__)) + "/../"
-assert (os.path.isfile (module_path  + "deepmd/libop_abi.so" )), "force module does not exist"
-op_module = tf.load_op_library(module_path + "deepmd/libop_abi.so")
+assert (os.path.isfile (module_path  + "deepmd/libop_abi.{}".format(ext) )), "force module does not exist"
+op_module = tf.load_op_library(module_path + "deepmd/libop_abi.{}".format(ext))
 
 # load grad of force module
 sys.path.append (module_path )
diff --git a/source/train/DeepEval.py b/source/train/DeepEval.py
@@ -1,14 +1,21 @@
 #!/usr/bin/env python3
-
+import platform
 import os,sys
 import numpy as np
 
 from deepmd.env import tf
 
 from tensorflow.python.framework import ops
+
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__))
-assert (os.path.isfile (os.path.join(module_path, "libop_abi.so"))), "op module does not exist"
-op_module = tf.load_op_library(os.path.join(module_path, "libop_abi.so"))
+assert (os.path.isfile (os.path.join(module_path, "libop_abi.{}".format(ext)))), "op module does not exist"
+op_module = tf.load_op_library(os.path.join(module_path, "libop_abi.{}".format(ext)))
 
 class DeepEval():
     """
diff --git a/source/train/DescrptLocFrame.py b/source/train/DescrptLocFrame.py
@@ -1,3 +1,4 @@
+import platform
 import os
 import numpy as np
 from deepmd.env import tf
@@ -8,9 +9,15 @@
 from deepmd.RunOptions import global_cvt_2_tf_float
 from deepmd.RunOptions import global_cvt_2_ener_float
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__)) + "/"
-assert (os.path.isfile (module_path  + "libop_abi.so" )), "op module does not exist"
-op_module = tf.load_op_library(module_path + "libop_abi.so")
+assert (os.path.isfile (module_path  + "libop_abi.{}".format(ext) )), "op module does not exist"
+op_module = tf.load_op_library(module_path + "libop_abi.{}".format(ext))
 
 class DescrptLocFrame () :
     def __init__(self, jdata):
diff --git a/source/train/DescrptSeA.py b/source/train/DescrptSeA.py
@@ -1,3 +1,4 @@
+import platform
 import os,sys,warnings
 import numpy as np
 from deepmd.env import tf
@@ -8,9 +9,15 @@
 from deepmd.RunOptions import global_cvt_2_tf_float
 from deepmd.RunOptions import global_cvt_2_ener_float
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__)) + "/"
-assert (os.path.isfile (module_path  + "libop_abi.so" )), "op module does not exist"
-op_module = tf.load_op_library(module_path + "libop_abi.so")
+assert (os.path.isfile (module_path  + "libop_abi.{}".format(ext) )), "op module does not exist"
+op_module = tf.load_op_library(module_path + "libop_abi.{}".format(ext))
 
 class DescrptSeA ():
     def __init__ (self, jdata):
diff --git a/source/train/DescrptSeAR.py b/source/train/DescrptSeAR.py
@@ -1,3 +1,4 @@
+import platform
 import os,sys,warnings
 import numpy as np
 from deepmd.env import tf
@@ -11,9 +12,15 @@
 from deepmd.DescrptSeA import DescrptSeA
 from deepmd.DescrptSeR import DescrptSeR
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__)) + "/"
-assert (os.path.isfile (module_path  + "libop_abi.so" )), "op module does not exist"
-op_module = tf.load_op_library(module_path + "libop_abi.so")
+assert (os.path.isfile (module_path  + "libop_abi.{}".format(ext) )), "op module does not exist"
+op_module = tf.load_op_library(module_path + "libop_abi.{}".format(ext))
 
 class DescrptSeAR ():
     def __init__ (self, jdata):
diff --git a/source/train/DescrptSeR.py b/source/train/DescrptSeR.py
@@ -1,4 +1,5 @@
 import os,warnings
+import platform
 import numpy as np
 from deepmd.env import tf
 from deepmd.common import ClassArg
@@ -8,9 +9,15 @@
 from deepmd.RunOptions import global_cvt_2_tf_float
 from deepmd.RunOptions import global_cvt_2_ener_float
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__)) + "/"
-assert (os.path.isfile (module_path  + "libop_abi.so" )), "op module does not exist"
-op_module = tf.load_op_library(module_path + "libop_abi.so")
+assert (os.path.isfile (module_path  + "libop_abi.{}".format(ext) )), "op module does not exist"
+op_module = tf.load_op_library(module_path + "libop_abi.{}".format(ext))
 
 class DescrptSeR ():
     def __init__ (self, jdata):
diff --git a/source/train/Model.py b/source/train/Model.py
@@ -1,4 +1,5 @@
 import os,sys,warnings
+import platform
 import numpy as np
 from deepmd.env import tf
 from collections import defaultdict
@@ -11,9 +12,15 @@
 from deepmd.RunOptions import global_cvt_2_tf_float
 from deepmd.RunOptions import global_cvt_2_ener_float
 
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__)) + "/"
-assert (os.path.isfile (module_path  + "libop_abi.so" )), "op module does not exist"
-op_module = tf.load_op_library(module_path + "libop_abi.so")
+assert (os.path.isfile (module_path  + "libop_abi.{}".format(ext) )), "op module does not exist"
+op_module = tf.load_op_library(module_path + "libop_abi.{}".format(ext))
 
 class Model() :
     model_type = 'ener'
diff --git a/source/train/Trainer.py b/source/train/Trainer.py
@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
 import os
+import platform
 import sys
 import time
 import shutil
@@ -24,9 +25,15 @@
 from tensorflow.python.client import timeline
 
 # load force module
+if platform.system() == "Windows":
+    ext = "dll"
+elif platform.system() == "Darwin":
+    ext = "dylib"
+else:
+    ext = "so"
 module_path = os.path.dirname(os.path.realpath(__file__)) + "/"
-assert (os.path.isfile (module_path  + "libop_abi.so" )), "op module does not exist"
-op_module = tf.load_op_library(module_path + "libop_abi.so")
+assert (os.path.isfile (module_path  + "libop_abi.{}".format(ext) )), "op module does not exist"
+op_module = tf.load_op_library(module_path + "libop_abi.{}".format(ext))
 
 # load grad of force module
 sys.path.append (module_path )
