Commit a4b49551 authored by lijian's avatar lijian
Browse files

change whl name for shca hca.


Signed-off-by: default avatarlijian <34831075+lijian0711@users.noreply.github.com>
parent 1a35d640
...@@ -268,10 +268,18 @@ fi ...@@ -268,10 +268,18 @@ fi
echo "Using Python: $(which python3)" echo "Using Python: $(which python3)"
python3 --version python3 --version
if [ "$USE_NVSHMEM" == "ON" ]; then if [ "$USE_NVSHMEM" == "ON" ]; then
if [ "$BUILD_SHCA" == "ON" ]; then
python setup.py bdist_wheel --shmem=nv --build_shca
else
python setup.py bdist_wheel --shmem=nv python setup.py bdist_wheel --shmem=nv
fi
fi fi
if [ "$USE_ROCSHMEM" == "ON" ]; then if [ "$USE_ROCSHMEM" == "ON" ]; then
if [ "$BUILD_SHCA" == "ON" ]; then
python setup.py bdist_wheel --shmem=rocm --build_shca
else
python setup.py bdist_wheel --shmem=rocm python setup.py bdist_wheel --shmem=rocm
fi
fi fi
echo "✅ Build complete:" echo "✅ Build complete:"
ls -lh dist/ ls -lh dist/
...@@ -15,6 +15,7 @@ if int(os.environ.get('ADD_GIT_VERSION', '0')) == 1: ...@@ -15,6 +15,7 @@ if int(os.environ.get('ADD_GIT_VERSION', '0')) == 1:
add_git_version = True add_git_version = True
shmem = None shmem = None
build_shca = False
other = [] other = []
for arg in sys.argv: for arg in sys.argv:
if arg.startswith("--shmem="): if arg.startswith("--shmem="):
...@@ -23,6 +24,8 @@ for arg in sys.argv: ...@@ -23,6 +24,8 @@ for arg in sys.argv:
shmem = "a" shmem = "a"
elif shmem == "nv": elif shmem == "nv":
shmem = "b" shmem = "b"
elif arg == "--build_shca":
build_shca = True
else: else:
other.append(arg) other.append(arg)
sys.argv = other sys.argv = other
...@@ -112,7 +115,7 @@ class BinaryDistribution(Distribution): ...@@ -112,7 +115,7 @@ class BinaryDistribution(Distribution):
if __name__ == '__main__': if __name__ == '__main__':
setuptools.setup( setuptools.setup(
name='deep_ep', name='deep_ep_shca' if build_shca else 'deep_ep',
version=get_deepep_version(), version=get_deepep_version(),
packages=setuptools.find_packages(include=['deep_ep']), packages=setuptools.find_packages(include=['deep_ep']),
include_package_data=True, include_package_data=True,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment