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): ...@@ -665,7 +665,10 @@ class CUDAOpBuilder(OpBuilder):
from torch.utils.hipify import hipify_python from torch.utils.hipify import hipify_python
#torch1.13 #torch1.13
import torch 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( hipify_python.hipify(
project_directory=os.getcwd(), project_directory=os.getcwd(),
output_directory=os.getcwd(), output_directory=os.getcwd(),
...@@ -679,6 +682,7 @@ class CUDAOpBuilder(OpBuilder): ...@@ -679,6 +682,7 @@ class CUDAOpBuilder(OpBuilder):
) )
else: else:
#torch1.10 #torch1.10
print("compile with torch1.10")
hipify_python.hipify( hipify_python.hipify(
project_directory=os.getcwd(), project_directory=os.getcwd(),
output_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