Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "167216afc8c6f1aa9046eec842374b61c11fdcf1"
Commit 0f3656b9 authored by aiss's avatar aiss
Browse files

modify error

parent 4acf0e01
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 1 deletion
op_builder/builder.py
View file @ 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