Skip to content

Commit

Permalink
Patch ProcessGroupGloo.hpp to address compilation issue on windows
Browse files Browse the repository at this point in the history
  • Loading branch information
HGuillemet committed Jun 8, 2024
1 parent 8b9b5ed commit 4760ec4
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions pytorch/cppbuild.sh
Original file line number Diff line number Diff line change
Expand Up @@ -182,6 +182,9 @@ TORCH_API std::ostream& operator<<(std::ostream& stream, const nn::Module& modul
' torch/csrc/api/include/torch/nn/module.h
sedinplace 's/char(\(.*\))/\1/g' torch/csrc/jit/serialization/pickler.h

# some windows header defines a macro named "interface"
sedinplace 's/const std::string& interface)/const std::string\& interface_name)/g' torch/csrc/distributed/c10d/ProcessGroupGloo.hpp

#USE_FBGEMM=0 USE_KINETO=0 USE_GLOO=0 USE_MKLDNN=0 \
"$PYTHON_BIN_PATH" setup.py build

Expand Down

0 comments on commit 4760ec4

Please sign in to comment.