From bfe51a87c5c459954fb010ccbf0002b975d85a9b Mon Sep 17 00:00:00 2001 From: rusty1s Date: Sat, 6 Apr 2024 09:36:32 +0000 Subject: [PATCH] update --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index d4532665..e1040f69 100644 --- a/setup.py +++ b/setup.py @@ -65,6 +65,7 @@ def build_extension(self, ext): '-DUSE_PYTHON=ON', f'-DWITH_CUDA={"ON" if WITH_CUDA else "OFF"}', f'-DCMAKE_LIBRARY_OUTPUT_DIRECTORY={extdir}', + f'-DCMAKE_RUNTIME_OUTPUT_DIRECTORY={extdir}', f'-DCMAKE_BUILD_TYPE={self.build_type}', f'-DCMAKE_PREFIX_PATH={torch.utils.cmake_prefix_path}', ]