Commit 5e176162 authored by gaoqiong's avatar gaoqiong
Browse files

增加DCU发版信息

parent 396acd66
Pipeline #1720 failed with stages
in 0 seconds
......@@ -24,44 +24,21 @@ docker run -it --name baichuan --shm-size=1024G -v /opt/hyhal:/opt/hyhal:ro --de
根据不同的需求下载不同的分支
- 提供2种源码编译方式(进入AutoAWQ目录):
```
1. 源码编译安装
export LD_LIBRARY_PATH=/usr/local/lib/python3.10/site-packages/torch/lib:$LD_LIBRARY_PATH
#源码编译安装
pip3 install e .
2. 编译成whl包安装
#编译成whl包安装
# 安装wheel
python3 setup.py bdist_wheel
cd dist && pip3 install autoawq*
```
## 验证
- python -c "import awq_ext; print(awq_ext.__version__)",版本号与官方版本同步,查询该软件的版本号,例如0.0.6;
## 参考资料
- [README](README.md)
- [https://github.com/casper-hansen/AutoAWQ_kernels](https://github.com/casper-hansen/AutoAWQ_kernels.git)
......@@ -43,7 +43,7 @@ def get_version_add(sha: Optional[str] = None) -> str:
version=''
autoawq_root = os.path.dirname(os.path.abspath(__file__))
add_version_path = os.path.join(os.path.join(autoawq_root, ""), "version.py")
add_version_path = os.path.join(os.path.join(autoawq_root, "awq_ext"), "__init__.py")
if add_git_version:
if sha != 'Unknown':
if sha is None:
......@@ -78,7 +78,7 @@ def get_version_add(sha: Optional[str] = None) -> str:
def get_version():
get_version_add()
version_file = 'version.py'
version_file = 'awq_ext/__init__.py'
with open(version_file, encoding='utf-8') as f:
exec(compile(f.read(), version_file, 'exec'))
return locals()['__dcu_version__']
......
__version__ = "0.0.6"
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