Commit 55ca26ec authored by lishen's avatar lishen
Browse files

fix


Signed-off-by: default avatarlishen <lishen@sugon.com>
parent a1382ed7
...@@ -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_NVSHMEM_API) #if defined(FORCE_DUSHMEM_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_NVSHMEM_API) #endif // defined(FORCE_DUSHMEM_API)
} }
__device__ __forceinline__ void __device__ __forceinline__ void
......
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
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