Commit ea76f44e authored by lijian6's avatar lijian6
Browse files

Add message for das release1.8 and update rocshmem for xdp.


Signed-off-by: lijian6's avatarlijian <lijian6@sugon.com>
parent 1110ef9d
#!/bin/bash #!/bin/bash
# set -eux set -eux
export amd_comgr_DIR=${ROCM_PATH}/lib64/cmake export amd_comgr_DIR=${ROCM_PATH}/lib64/cmake
llvm15_path=${ROCM_PATH}/llvm/lib/clang/15.0.0 llvm15_path=${ROCM_PATH}/llvm/lib/clang/15.0.0
...@@ -184,7 +184,8 @@ for src in "${SOURCES[@]}"; do ...@@ -184,7 +184,8 @@ for src in "${SOURCES[@]}"; do
done done
# 链接阶段 # 链接阶段
OUTPUT="deep_ep/deep_ep_cpp.cpython-310-x86_64-linux-gnu.so" ext_suffix=$(python3 -c 'import sysconfig; print(sysconfig.get_config_var("EXT_SUFFIX"))')
OUTPUT="deep_ep/deep_ep_cpp$ext_suffix"
# 检查是否需要重新链接 # 检查是否需要重新链接
need_link=false need_link=false
......
...@@ -2,6 +2,7 @@ import os, sys ...@@ -2,6 +2,7 @@ import os, sys
import subprocess import subprocess
import setuptools import setuptools
import subprocess import subprocess
import sysconfig
from typing import Optional from typing import Optional
import subprocess import subprocess
...@@ -110,6 +111,6 @@ if __name__ == '__main__': ...@@ -110,6 +111,6 @@ if __name__ == '__main__':
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,
package_data={"deep_ep": ["deep_ep_cpp.cpython-310-x86_64-linux-gnu.so"]}, package_data={"deep_ep": [f"deep_ep_cpp{sysconfig.get_config_var('EXT_SUFFIX')}"]},
zip_safe=False, zip_safe=False,
) )
rocshmem @ d8a4cb56
Subproject commit 9735e605803e3c238805c103a7d3dfa8da4c4ca2 Subproject commit d8a4cb56d62694d1ac624f36a3db4325dd7c4d89
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