File tree Expand file tree Collapse file tree 2 files changed +3
-1
lines changed Expand file tree Collapse file tree 2 files changed +3
-1
lines changed Original file line number Diff line number Diff 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/...",
Original file line number Diff line number Diff line change 99
1010import torch
1111from executorch .backends .cuda .cuda_partitioner import CudaPartitioner
12+ from executorch .backends .cuda .cuda_backend import CudaBackend
1213from executorch .exir import EdgeCompileConfig , to_edge_transform_and_lower
1314from 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 (
You can’t perform that action at this time.
0 commit comments