up

Author: metascroy

.github/workflows/_unittest.yml

@@ -43,7 +43,7 @@ jobs:
   macos:
     uses: pytorch/test-infra/.github/workflows/macos_job.yml@main
     with:
-      runner: macos-15-xlarge
+      runner: macos-m1-stable
       python-version: '3.11'
       submodules: 'recursive'
       ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}

backends/apple/coreml/test/test_torch_ops.py

@@ -2,10 +2,7 @@
 #
 # Please refer to the license found in the LICENSE file in the root directory of the source tree.
 
-<<<<<<< HEAD
 import platform
-=======
->>>>>>> 921dafe9b (up)
 import sys
 import unittest
 

exir/program/_program.py

@@ -1123,6 +1123,9 @@ def _gen_edge_manager_for_partitioners(
     ops_set_to_not_decompose_by_program = {}
     edge_programs: Dict[str, ExportedProgram] = {}
     for name, program in aten_programs.items():
+        # Functionalize program before asking partitioners to preserve ops
+        program = program.run_decompositions({})
+
         if partitioner is not None:
             # preserve all ops listed by all partitioners first
             all_ops_no_decomp = set()