[MPSInductor] Fix noop codegen (#151224)

malfet · pytorchmergebot · commit 184ac8c7f76d · 2025-04-14T16:38:47.000Z
By adding `pass` in front of the comment for fake set_device call Which fixes `TestGPU.test_zero_element_mutation_mps`, which previously failed with ``` torch._inductor.exc.InductorError: RuntimeError: Failed to import /var/folders/sc/2thx6_x95h7_h9qs8s48yh140000gn/T/tmp2emka_sx/7k/c7kmnwhb363ysalhewglr3cwtej6tiz3t4ppqa4bvhubaokmlprw.py IndentationError: expected an indented block after 'with' statement on line 38 (c7kmnwhb363ysalhewglr3cwtej6tiz3t4ppqa4bvhubaokmlprw.py, line 40) ``` Pull Request resolved: #151224 Approved by: https://github.com/Skylion007, https://github.com/jansel, https://github.com/dcci
diff --git a/test/inductor/test_mps_basic.py b/test/inductor/test_mps_basic.py
@@ -242,6 +242,7 @@ def fn(a):
     "test_views6",
     "test_views7",
     "test_zero_dim_reductions",
+    "test_zero_element_mutation",
 ]:
     setattr(MPSBasicTests, test_name, getattr(CommonTemplate, test_name))
 
diff --git a/torch/_inductor/codegen/mps_device_op_overrides.py b/torch/_inductor/codegen/mps_device_op_overrides.py
@@ -10,7 +10,7 @@ def device_guard(self, device_idx: int) -> str:
 
     def set_device(self, device_idx: int) -> str:
         assert device_idx == 0
-        return "# MPS set device"
+        return "pass  # MPS set device"
 
 
 register_device_op_overrides("mps", MPSDeviceOpOverrides())
