Skip to content

GitLab

Commit d8aab4e1 authored by lijian's avatar lijian
Browse files

Support compile for torch 2.1. 


Signed-off-by: default avatarlijian <34831075+lijian0711@users.noreply.github.com>
parent 905e9e87
Pipeline #3697 failed with stages
in 0 seconds
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
setup.py
View file @ d8aab4e1
...@@ -40,10 +40,10 @@ def get_version_add(sha: Optional[str] = None) -> str: ...@@ -40,10 +40,10 @@ def get_version_add(sha: Optional[str] = None) -> str:
if sha != 'Unknown': if sha != 'Unknown':
if sha is None: if sha is None:
sha = subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=deepep_root).decode('ascii').strip() sha = subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=deepep_root).decode('ascii').strip()
if (major, minor) >= ('2', '4'): if (major, minor) >= ('2', '1'):
version = 'das.opt1.' + sha[:7] + shmem version = 'das.opt1.' + sha[:7] + shmem
else: else:
if (major, minor) >= ('2', '4'): if (major, minor) >= ('2', '1'):
version = 'das.opt1' version = 'das.opt1'
if os.getenv("ROCM_PATH"): 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