Add pyproject.toml with build system dependencies for PEP-517 installs

CHANGELOG.md

@@ -19,6 +19,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/).
 
 ### Fixed
 
+- Fixed install with `--use-pep517` by specifying build-system dependencies. ([#3758](https://github.com/horovod/horovod/pull/3758))
 - Fixed memory leak in MPI_GPUAllgather. ([#3727](https://github.com/horovod/horovod/pull/3727))
 
 

_custom_build/backend.py

@@ -0,0 +1,13 @@
+from setuptools import build_meta as _orig
+
+prepare_metadata_for_build_wheel = _orig.prepare_metadata_for_build_wheel
+build_wheel = _orig.build_wheel
+build_sdist = _orig.build_sdist
+
+
+def get_requires_for_build_wheel(self, config_settings=None):
+ return _orig.get_requires_for_build_wheel(config_settings) + ["tensorflow"]
+
+
+def get_requires_for_build_sdist(self, config_settings=None):
+ return _orig.get_requires_for_build_sdist(config_settings) + ["tensorflow"]

pyproject.toml

@@ -0,0 +1,5 @@
+[build-system]
+requires = ["packaging", "setuptools"]
+build-backend = "backend"
+backend-path = ["_custom_build"]
+

setup.py

@@ -28,7 +28,7 @@
 from setuptools import setup, Extension, find_packages
 from setuptools.command.build_ext import build_ext
 
-from horovod import __version__
+__version__ = "0.26.1"
 
 _FRAMEWORK_METADATA_FILE = 'horovod/metadata.json'