Merge pull request #499 from JiabinYang/deemd2paddle

Deepmd2paddle
deepmodeling · Apr 13, 2021 · 9426f1c · 9426f1c
2 parents 1a06aa4 + c24e33e
commit 9426f1c
Show file tree

Hide file tree

Showing 9 changed files with 2,378 additions and 0 deletions.
diff --git a/deepmd/load_paddle_op.py b/deepmd/load_paddle_op.py
@@ -0,0 +1,17 @@
+from paddle.utils.cpp_extension import CppExtension, CUDAExtension, setup
+
+setup(
+    name='paddle_ops',
+    ext_modules=CppExtension(
+        sources=['../source/op/paddle_ops/srcs/pd_prod_env_mat_multi_devices_cpu.cc',
+        '../source/op/paddle_ops/srcs/pd_prod_env_mat_multi_devices_cuda.cc',
+        '../source/op/paddle_ops/srcs/pd_prod_force_se_a_multi_devices_cpu.cc',
+        '../source/op/paddle_ops/srcs/pd_prod_force_se_a_multi_devices_cuda.cc',
+        '../source/op/paddle_ops/srcs/pd_prod_virial_se_a_multi_devices_cpu.cc',
+        '../source/op/paddle_ops/srcs/pd_prod_virial_se_a_multi_devices_cuda.cc'],
+        include_dirs=["../source/lib/include/","/usr/local/cuda-10.1/targets/x86_64-linux/include/"],
+        library_dirs=["../build/lib/", "/usr/local/cuda-10.1/lib64"],
+        extra_link_args=["-ldeepmd","-lcudart"]
+    )
+)
+