Skip to content

GitLab

Commit e3475937 authored by zhuwenwen's avatar zhuwenwen
Browse files

modify dcu_version

parent 4b5498c1
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
fastfold/__init__.py
View file @ e3475937
from .version import __version__, __dcu_version__
VERSION = "0.1.0-beta" VERSION = "0.1.0-beta"
\ No newline at end of file
setup.py
View file @ e3475937
...@@ -133,7 +133,7 @@ def get_abi(): ...@@ -133,7 +133,7 @@ def get_abi():
def get_version_add(sha: Optional[str] = None) -> str: def get_version_add(sha: Optional[str] = None) -> str:
fastfold_root = os.path.dirname(os.path.abspath(__file__)) fastfold_root = os.path.dirname(os.path.abspath(__file__))
add_version_path = "version.py" add_version_path = os.path.join(os.path.join(fastfold_root, "fastfold"), "version.py")
if sha != 'Unknown': if sha != 'Unknown':
if sha is None: if sha is None:
sha = get_sha(fastfold_root) sha = get_sha(fastfold_root)
...@@ -142,7 +142,7 @@ def get_version_add(sha: Optional[str] = None) -> str: ...@@ -142,7 +142,7 @@ def get_version_add(sha: Optional[str] = None) -> str:
# abi version # abi version
version += "." + get_abi() version += "." + get_abi()
# dtk version # dtk version
if os.getenv("ROCM_PATH"): if os.getenv("ROCM_PATH"):
rocm_path = os.getenv('ROCM_PATH', "") rocm_path = os.getenv('ROCM_PATH', "")
rocm_version_path = os.path.join(rocm_path, '.info', "rocm_version") rocm_version_path = os.path.join(rocm_path, '.info', "rocm_version")
......
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