From 869ac2c0f65cd7e7cbf710c345a48190d9c59b25 Mon Sep 17 00:00:00 2001 From: Stefan Doerr Date: Tue, 21 Jan 2025 17:17:54 +0200 Subject: [PATCH] renamed extensions file to attempsolve issues with Windows compilation --- setup.py | 2 +- .../extensions/{extensions.cpp => torchmdnet_extensions.cpp} | 0 2 files changed, 1 insertion(+), 1 deletion(-) rename torchmdnet/extensions/{extensions.cpp => torchmdnet_extensions.cpp} (100%) diff --git a/setup.py b/setup.py index 81faa3d8d..f56277adb 100644 --- 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 b/torchmdnet/extensions/torchmdnet_extensions.cpp similarity index 100% rename from torchmdnet/extensions/extensions.cpp rename to torchmdnet/extensions/torchmdnet_extensions.cpp