"vllm/vscode:/vscode.git/clone" did not exist on "ff334ca1cd92c41cc79e9dead91de40b87601daf"
Commit 52c3dd59 authored by zhuwenwen's avatar zhuwenwen
Browse files

add git version for internal testing

parent 04633c47
......@@ -19,6 +19,9 @@ import subprocess
from pathlib import Path
if int(os.environ.get('ADD_GIT_VERSION', '0')) == 1:
add_git_version = True
def load_module_from_path(module_name, path):
spec = importlib.util.spec_from_file_location(module_name, path)
module = importlib.util.module_from_spec(spec)
......@@ -309,11 +312,23 @@ def find_version(filepath: str) -> str:
raise RuntimeError("Unable to find version string.")
def get_sha(root: Union[str, Path]) -> str:
try:
return subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=root).decode('ascii').strip()
except Exception:
return 'Unknown'
def get_version_add(sha: Optional[str] = None) -> str:
vllm_root = os.path.dirname(os.path.abspath(__file__))
add_version_path = os.path.join(os.path.join(vllm_root, "vllm"), "version.py")
version = 'das.opt1'
if add_git_version:
if sha != 'Unknown':
if sha is None:
sha = get_sha(vllm_root)
version = 'das.opt1.' + sha[:7]
else:
version = 'das.opt1'
# dtk version
if os.getenv("ROCM_PATH"):
......
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