Fix macOS build with USE_MPS=OFF (#156932)

pytorchbot · XuehaiPan · web-flow · commit 90d16ee8004b · 2025-07-07T14:59:26.000-07:00
Fix macOS build with `USE_MPS=OFF` (#156847) Pull Request resolved: #156847 Approved by: https://github.com/angelayi (cherry picked from commit 455dfd2) Co-authored-by: Xuehai Pan <XuehaiPan@pku.edu.cn>
diff --git a/torch/csrc/inductor/aoti_runner/pybind.cpp b/torch/csrc/inductor/aoti_runner/pybind.cpp
@@ -134,7 +134,8 @@ void initAOTIRunnerBindings(PyObject* module) {
           &AOTIModelContainerRunnerXpu::free_inactive_constant_buffer);
 
 #endif
-#if defined(__APPLE__) && !(defined(FBCODE_CAFFE2) || defined(OVRSOURCE))
+#if defined(USE_MPS) && defined(__APPLE__) && \
+    !(defined(FBCODE_CAFFE2) || defined(OVRSOURCE))
   py::class_<AOTIModelContainerRunnerMps>(m, "AOTIModelContainerRunnerMps")
       .def(py::init<const std::string&, int>())
       .def(
