renamed extensions file to attempsolve issues with Windows compilation

torchmd · Jan 21, 2025 · 869ac2c · 869ac2c
1 parent b90b49d
commit 869ac2c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -43,7 +43,7 @@ def set_torch_cuda_arch_list():
 ExtensionType = CppExtension if not use_cuda else CUDAExtension
 extensions = ExtensionType(
     name='torchmdnet.extensions.torchmdnet_extensions',
-    sources=[os.path.join(extension_root, "extensions.cpp")] + neighbor_sources,
+    sources=[os.path.join(extension_root, "torchmdnet_extensions.cpp")] + neighbor_sources,
     include_dirs=include_paths(),
     define_macros=[('WITH_CUDA', 1)] if use_cuda else [],
 )

diff --git a/torchmdnet/extensions/extensions.cpp → ...dnet/extensions/torchmdnet_extensions.cpp b/torchmdnet/extensions/extensions.cpp → ...dnet/extensions/torchmdnet_extensions.cpp