Fix C++ extension for Windows (#1371)

* Fix install location and suffix for Windows

Fix C++ extension for Windows (#1371)
* Fix install location and suffix for Windows
f5ac116f · moto · GitHub · 7b85f1c3 · f5ac116f
Unverified Commit f5ac116f authored Mar 08, 2021 by moto Committed by GitHub Mar 08, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 1 deletion

torchaudio/csrc/CMakeLists.txt torchaudio/csrc/CMakeLists.txt +8 -1

No files found.
--- a/torchaudio/csrc/CMakeLists.txt
+++ b/torchaudio/csrc/CMakeLists.txt
@@ -74,6 +74,9 @@ if (BUILD_TORCHAUDIO_PYTHON_EXTENSION)
    )

  set_target_properties(_torchaudio PROPERTIES PREFIX "")
+  if (MSVC)
+    set_target_properties(_torchaudio PROPERTIES SUFFIX ".pyd")
+  endif(MSVC)

  if (APPLE)
    # https://github.com/facebookarchive/caffe2/issues/854#issuecomment-364538485
@@ -116,5 +119,9 @@ if (BUILD_TORCHAUDIO_PYTHON_EXTENSION)
    ${ADDITIONAL_ITEMS}
    )

-  install(TARGETS _torchaudio LIBRARY DESTINATION ${CMAKE_INSTALL_PREFIX})
+  install(
+    TARGETS _torchaudio
+    LIBRARY DESTINATION .
+    RUNTIME DESTINATION .  # For Windows
+    )
 endif()