Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
rusty1s committed Apr 6, 2024
1 parent 0a6d0a9 commit bfe51a8
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -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}',
]
Expand Down

0 comments on commit bfe51a8

Please sign in to comment.