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
55ca26ec
Commit
55ca26ec
authored
Jan 23, 2026
by
lishen
Browse files
fix
Signed-off-by:
lishen
<
lishen@sugon.com
>
parent
a1382ed7
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
41 additions
and
3 deletions
+41
-3
csrc/kernels/internode_ll.cu
csrc/kernels/internode_ll.cu
+2
-2
deep_ep/version.py
deep_ep/version.py
+38
-0
third-party/rocshmem
third-party/rocshmem
+1
-1
No files found.
csrc/kernels/internode_ll.cu
View file @
55ca26ec
...
@@ -84,7 +84,7 @@ __device__ __forceinline__ void
...
@@ -84,7 +84,7 @@ __device__ __forceinline__ void
internode_ll_putmem_nbi
(
void
*
dst_ptr
,
void
*
src_ptr
,
internode_ll_putmem_nbi
(
void
*
dst_ptr
,
void
*
src_ptr
,
int
num_ranks
,
int
dst_rank
,
int
expert_idx
,
int
num_ranks
,
int
dst_rank
,
int
expert_idx
,
int
msg_bytes
)
{
int
msg_bytes
)
{
#if defined(FORCE_
NV
SHMEM_API)
#if defined(FORCE_
DU
SHMEM_API)
internode
::
shmemx_int8_put_nbi_warp
(
internode
::
shmemx_int8_put_nbi_warp
(
reinterpret_cast
<
signed
char
*>
(
dst_ptr
),
reinterpret_cast
<
signed
char
*>
(
src_ptr
),
reinterpret_cast
<
signed
char
*>
(
dst_ptr
),
reinterpret_cast
<
signed
char
*>
(
src_ptr
),
msg_bytes
,
dst_rank
);
msg_bytes
,
dst_rank
);
...
@@ -98,7 +98,7 @@ internode_ll_putmem_nbi(void* dst_ptr, void* src_ptr,
...
@@ -98,7 +98,7 @@ internode_ll_putmem_nbi(void* dst_ptr, void* src_ptr,
reinterpret_cast
<
signed
char
*>
(
dst_ptr
),
reinterpret_cast
<
signed
char
*>
(
src_ptr
),
reinterpret_cast
<
signed
char
*>
(
dst_ptr
),
reinterpret_cast
<
signed
char
*>
(
src_ptr
),
msg_bytes
,
(
expert_idx
+
1
)
*
num_ranks
+
dst_rank
,
dst_rank
);
msg_bytes
,
(
expert_idx
+
1
)
*
num_ranks
+
dst_rank
,
dst_rank
);
#endif
#endif
#endif // defined(FORCE_
NV
SHMEM_API)
#endif // defined(FORCE_
DU
SHMEM_API)
}
}
__device__
__forceinline__
void
__device__
__forceinline__
void
...
...
deep_ep/version.py
0 → 100644
View file @
55ca26ec
try
:
__version__
=
"1.0.0"
__version_tuple__
=
(
1
,
0
,
0
)
__hcu_version__
=
f
'1.0.0+das.opt1.dtk25042'
from
deep_ep.version
import
__version__
,
__version_tuple__
,
__hcu_version__
except
Exception
as
e
:
import
warnings
warnings
.
warn
(
f
"Failed to read commit hash:
\n
+ str(e)"
,
RuntimeWarning
,
stacklevel
=
2
)
__version__
=
"dev"
__version_tuple__
=
(
0
,
0
,
__version__
)
def
_prev_minor_version_was
(
version_str
):
'''Check whether a given version matches the previous minor version.
Return True if version_str matches the previous minor version.
For example - return True if the current version is 0.7.4 and the
supplied version_str is '0.6'.
Used for --show-hidden-metrics-for-version.
'''
# Match anything if this is a dev tree
if
__version_tuple__
[
0
:
2
]
==
(
0
,
0
):
return
True
# Note - this won't do the right thing when we release 1.0!
# assert __version_tuple__[0] == 0
assert
isinstance
(
__version_tuple__
[
1
],
int
)
return
version_str
==
f
"
{
__version_tuple__
[
0
]
}
.
{
__version_tuple__
[
1
]
-
1
}
"
def
_prev_minor_version
():
'''For the purpose of testing, return a previous minor version number.'''
# In dev tree, this will return "0.-1", but that will work fine"
assert
isinstance
(
__version_tuple__
[
1
],
int
)
return
f
"
{
__version_tuple__
[
0
]
}
.
{
__version_tuple__
[
1
]
-
1
}
"
rocshmem
@
9735e605
Subproject commit
cb2af8b8be114cd5e640e46446325301df93eedb
Subproject commit
9735e605803e3c238805c103a7d3dfa8da4c4ca2
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