File tree Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Expand file tree Collapse file tree 2 files changed +5
-1
lines changed Original file line number Diff line number Diff line change @@ -11,7 +11,7 @@ def _init_extension():
1111 warnings .warn ('torchaudio C++ extension is not available.' )
1212 return
1313
14- suffix = 'dll ' if os .name == 'nt' else 'so'
14+ suffix = 'pyd ' if os .name == 'nt' else 'so'
1515 path = Path (__file__ ).parent / 'lib' / f'libtorchaudio.{ suffix } '
1616 # In case `torchaudio` is deployed with `pex` format, this file does not exist.
1717 # In this case, we expect that `libtorchaudio` is available somewhere
Original file line number Diff line number Diff line change @@ -90,6 +90,10 @@ if(USE_CUDA)
9090 )
9191endif ()
9292
93+ if (MSVC )
94+ set_target_properties (libtorchaudio PROPERTIES SUFFIX ".pyd" )
95+ endif (MSVC )
96+
9397install (
9498 TARGETS libtorchaudio
9599 LIBRARY DESTINATION lib
You can’t perform that action at this time.
0 commit comments