Skip to content

Commit 72a2761

Browse files
larryliu0820facebook-github-bot
authored andcommitted
Fix cuda export test failures from #14715
Summary: Fix Reviewed By: abhinaykukkadapu Differential Revision: D83765135
1 parent deb42f2 commit 72a2761

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

backends/cuda/TARGETS

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@ runtime.python_library(
66
name = "cuda_backend",
77
srcs = [
88
"cuda_backend.py",
9+
"replace_slice_copy_with_slice.py",
910
],
1011
visibility = [
1112
"//executorch/...",

backends/cuda/tests/test_cuda_export.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99

1010
import torch
1111
from executorch.backends.cuda.cuda_partitioner import CudaPartitioner
12+
from executorch.backends.cuda.cuda_backend import CudaBackend
1213
from executorch.exir import EdgeCompileConfig, to_edge_transform_and_lower
1314
from torch.export import export
1415

@@ -30,7 +31,7 @@ def _export_to_cuda_with_lower(
3031
exported_program = export(module, inputs, strict=True)
3132

3233
# Create partitioner and compile specs
33-
partitioner = CudaPartitioner([])
34+
partitioner = CudaPartitioner([CudaBackend.generate_method_name_compile_spec("forward")])
3435

3536
# Use to_edge_transform_and_lower for complete pipeline
3637
edge_program_manager = to_edge_transform_and_lower(

0 commit comments

Comments
 (0)