Commit 7fae23b4 authored by lijian6's avatar lijian6
Browse files

Modify setup.py for new whl name


Signed-off-by: lijian6's avatarlijian <lijian6@sugon.com>
parent 03470cd3
__dcu_version__ = '0.29.0+das1.2.git02afa54.abi0.dtk2404'
__dcu_version__ = '0.29.0+das.opt1.dtk24040'
......@@ -254,33 +254,33 @@ install_requires = [
version_range_max = max(sys.version_info[1], 10) + 1
def get_sha(root_dir: Union[str, Path]) -> str:
try:
return subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=root_dir).decode('ascii').strip()
except Exception:
return 'Unknown'
def get_abi():
try:
command = "echo '#include <string>' | gcc -x c++ -E -dM - | fgrep _GLIBCXX_USE_CXX11_ABI"
result = subprocess.run(command, shell=True, capture_output=True, text=True)
output = result.stdout.strip()
abi = "abi" + output.split(" ")[-1]
return abi
except Exception:
return 'abiUnknown'
#def get_sha(root_dir: Union[str, Path]) -> str:
# try:
# return subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=root_dir).decode('ascii').strip()
# except Exception:
# return 'Unknown'
#
#def get_abi():
# try:
# command = "echo '#include <string>' | gcc -x c++ -E -dM - | fgrep _GLIBCXX_USE_CXX11_ABI"
# result = subprocess.run(command, shell=True, capture_output=True, text=True)
# output = result.stdout.strip()
# abi = "abi" + output.split(" ")[-1]
# return abi
# except Exception:
# return 'abiUnknown'
def get_version_add(sha: Optional[str] = None) -> str:
version=''
root_dir = os.path.dirname(os.path.abspath(__file__))
add_version_path = os.path.join(root_dir, "DIFFUSERS_VERSION_NUMBER")
if sha != 'Unknown':
if sha is None:
sha = get_sha(root_dir)
version = 'git' + sha[:7]
#if sha != 'Unknown':
# if sha is None:
# sha = get_sha(root_dir)
# version = 'git' + sha[:7]
# abi
version += "." + get_abi()
## abi
#version += "." + get_abi()
# dtk version
if os.getenv("ROCM_PATH"):
......@@ -288,13 +288,13 @@ def get_version_add(sha: Optional[str] = None) -> str:
rocm_version_path = os.path.join(rocm_path, '.info', "rocm_version")
with open(rocm_version_path, 'r',encoding='utf-8') as file:
lines = file.readlines()
rocm_version=lines[0][:-2].replace(".", "")
version += ".dtk" + rocm_version
rocm_version=lines[0][:-1].replace(".", "")
version += "dtk" + rocm_version
lines=[]
with open(add_version_path, 'r',encoding='utf-8') as file:
lines = file.readlines()
lines[0] = "__dcu_version__ = '0.29.0+das1.2.{}'\n".format(version)
lines[0] = "__dcu_version__ = '0.29.0+das.opt1.{}'\n".format(version)
with open(add_version_path, encoding="utf-8",mode="w") as file:
file.writelines(lines)
file.close()
......@@ -302,7 +302,7 @@ def get_version_add(sha: Optional[str] = None) -> str:
init_path=os.path.join(root_dir, "src/diffusers/__init__.py")
with open(init_path, 'r',encoding='utf-8') as file:
lines = file.readlines()
lines[1] = "__dcu_version__ = '0.29.0+das1.2.{}'\n".format(version)
lines[1] = "__dcu_version__ = '0.29.0+das.opt1.{}'\n".format(version)
lines[818] = " extra_objects={\"__version__\": __version__, \"__dcu_version__\": __dcu_version__},\n"
with open(init_path, encoding="utf-8",mode="w") as file:
file.writelines(lines)
......
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