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
echo "Using Python: $(which python3)"
python3 --version
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
fi
fi
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
fi
fi
echo "✅ Build complete:"
ls -lh dist/
......@@ -15,6 +15,7 @@ if int(os.environ.get('ADD_GIT_VERSION', '0')) == 1:
add_git_version = True
shmem = None
build_shca = False
other = []
for arg in sys.argv:
if arg.startswith("--shmem="):
......@@ -23,6 +24,8 @@ for arg in sys.argv:
shmem = "a"
elif shmem == "nv":
shmem = "b"
elif arg == "--build_shca":
build_shca = True
else:
other.append(arg)
sys.argv = other
......@@ -112,7 +115,7 @@ class BinaryDistribution(Distribution):
if __name__ == '__main__':
setuptools.setup(
name='deep_ep',
name='deep_ep_shca' if build_shca else 'deep_ep',
version=get_deepep_version(),
packages=setuptools.find_packages(include=['deep_ep']),
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