Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
OpenDAS
DeepEP
Commits
a4b49551
Commit
a4b49551
authored
May 23, 2026
by
lijian
Browse files
change whl name for shca hca.
Signed-off-by:
lijian
<
34831075+lijian0711@users.noreply.github.com
>
parent
1a35d640
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
14 additions
and
3 deletions
+14
-3
build.sh
build.sh
+10
-2
setup.py
setup.py
+4
-1
No files found.
build.sh
View file @
a4b49551
...
@@ -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/
setup.py
View file @
a4b49551
...
@@ -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
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment