Skip to content

GitLab

Commit d8669b08 authored by aiss's avatar aiss
Browse files

modify dtk path

parent 7d1a83a9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
deepspeed/ops/op_builder/builder.py
View file @ d8669b08
...@@ -190,7 +190,7 @@ class OpBuilder(ABC): ...@@ -190,7 +190,7 @@ class OpBuilder(ABC):
if OpBuilder.is_rocm_pytorch(): if OpBuilder.is_rocm_pytorch():
from torch.utils.cpp_extension import ROCM_HOME from torch.utils.cpp_extension import ROCM_HOME
#with open('/opt/rocm/.info/version-dev', 'r') as file: #with open('/opt/rocm/.info/version-dev', 'r') as file:
with open('/public/home/aiss/2022/dtk-21.10.1/.info/version-dev', 'r') as file: with open('/opt/dtk-22.04/.info/version-dev', 'r') as file:
ROCM_VERSION_DEV_RAW = file.read() ROCM_VERSION_DEV_RAW = file.read()
ROCM_MAJOR = ROCM_VERSION_DEV_RAW.split('.')[0] ROCM_MAJOR = ROCM_VERSION_DEV_RAW.split('.')[0]
ROCM_MINOR = ROCM_VERSION_DEV_RAW.split('.')[1] ROCM_MINOR = ROCM_VERSION_DEV_RAW.split('.')[1]
......
op_builder/builder.py
View file @ d8669b08
...@@ -190,7 +190,7 @@ class OpBuilder(ABC): ...@@ -190,7 +190,7 @@ class OpBuilder(ABC):
if OpBuilder.is_rocm_pytorch(): if OpBuilder.is_rocm_pytorch():
from torch.utils.cpp_extension import ROCM_HOME from torch.utils.cpp_extension import ROCM_HOME
#with open('/opt/rocm/.info/version-dev', 'r') as file: #with open('/opt/rocm/.info/version-dev', 'r') as file:
with open('/public/home/aiss/2022/dtk-21.10.1/.info/version-dev', 'r') as file: with open('/opt/dtk-22.04/.info/version-dev', 'r') as file:
ROCM_VERSION_DEV_RAW = file.read() ROCM_VERSION_DEV_RAW = file.read()
ROCM_MAJOR = ROCM_VERSION_DEV_RAW.split('.')[0] ROCM_MAJOR = ROCM_VERSION_DEV_RAW.split('.')[0]
ROCM_MINOR = ROCM_VERSION_DEV_RAW.split('.')[1] ROCM_MINOR = ROCM_VERSION_DEV_RAW.split('.')[1]
......
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