Commit 6a707da5 authored by aiss's avatar aiss
Browse files

Merge branch 'ds-v0.8.2-rocm' into 'main'

modify error

See merge request aicomponent/deepspeed!2
parents 899b52ce 0f3656b9
......@@ -665,7 +665,10 @@ class CUDAOpBuilder(OpBuilder):
from torch.utils.hipify import hipify_python
#torch1.13
import torch
if (torch_version.startswith('torch1.13')):
torch_version = torch.__version__
print("torch version: ", torch.__version__)
if (torch_version.startswith('1.13')):
print("compile with torch1.13!")
hipify_python.hipify(
project_directory=os.getcwd(),
output_directory=os.getcwd(),
......@@ -679,6 +682,7 @@ class CUDAOpBuilder(OpBuilder):
)
else:
#torch1.10
print("compile with torch1.10")
hipify_python.hipify(
project_directory=os.getcwd(),
output_directory=os.getcwd(),
......
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