Skip to content

Commit 4732087

Browse files
committed
solve cuda backend dependency issue
1 parent 42face6 commit 4732087

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backends/cuda/cuda_partitioner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
from typing import Callable, Dict, final, List, Optional, Tuple
1010

1111
import torch
12-
from executorch.backends.aoti.cuda.cuda_backend import CudaBackend # usort: skip
12+
from executorch.backends.cuda.cuda_backend import CudaBackend # usort: skip
1313
from executorch.exir.backend.compile_spec_schema import CompileSpec
1414
from executorch.exir.backend.partitioner import (
1515
DelegationSpec,

export_aoti.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from typing import Any, Dict, Tuple
2424

2525
import torch
26-
from executorch.backends.aoti.cuda.cuda_partitioner import CudaPartitioner
26+
from executorch.backends.cuda.cuda_partitioner import CudaPartitioner
2727

2828
# from executorch.backends.xnnpack.partition.xnnpack_partitioner import XnnpackPartitioner
2929
from executorch.exir import to_edge, to_edge_transform_and_lower

0 commit comments

Comments
 (0)