Skip to content

Commit

Permalink
Fix builds for PyTorch on Windows
Browse files Browse the repository at this point in the history
  • Loading branch information
saudet committed Oct 24, 2024
1 parent d336eaf commit c0dec78
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions pytorch/cppbuild.sh
Original file line number Diff line number Diff line change
Expand Up @@ -184,6 +184,7 @@ sedinplace 's/ build_deps()/ build_deps(); sys.exit()/g' setup.py
sedinplace 's/AND NOT DEFINED ENV{CUDAHOSTCXX}//g' cmake/public/cuda.cmake
sedinplace 's/CMAKE_CUDA_FLAGS "/CMAKE_CUDA_FLAGS " --use-local-env /g' CMakeLists.txt

sedinplace '/pycore_opcode.h/d' torch/csrc/dynamo/cpython_defs.c functorch/csrc/dim/dim*
sedinplace 's/using ExpandingArrayDouble/public: using ExpandingArrayDouble/g' ./torch/csrc/api/include/torch/nn/options/pooling.h

# allow setting the build directory and passing CUDA options
Expand Down

0 comments on commit c0dec78

Please sign in to comment.