Improve behavior on OS X (#20)

* Adding a guard to check if the Process class has the cpu_affinity function.

* Import the inspect class.

* Remove debug code.

Author: bvanessen

hpc_launcher/schedulers/local.py

@@ -1,3 +1,16 @@
+# Copyright (c) 2014-2024, Lawrence Livermore National Security, LLC.
+# Produced at the Lawrence Livermore National Laboratory.
+# Written by the LBANN Research Team (B. Van Essen, et al.) listed in
+# the CONTRIBUTORS file. See the top-level LICENSE file for details.
+#
+# LLNL-CODE-697807.
+# All rights reserved.
+#
+# This file is part of LBANN: Livermore Big Artificial Neural Network
+# Toolkit. For details, see http://software.llnl.gov/LBANN or
+# https://github.com/LBANN and https://github.com/LLNL/LBANN.
+#
+# SPDX-License-Identifier: (Apache-2.0)
 from hpc_launcher.schedulers.scheduler import Scheduler
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Optional

hpc_launcher/torch/__init__.py

@@ -12,14 +12,18 @@
 #
 # SPDX-License-Identifier: (Apache-2.0)
 from psutil import Process
+import inspect
 
-# Save affinity before importing torch
-affinity = Process().cpu_affinity()
+affinity = None
+if (hasattr(Process, 'cpu_affinity') and inspect.isfunction(Process.cpu_affinity)):
+    # Save affinity before importing torch
+    affinity = Process().cpu_affinity()
 
 import torch
 
-# Restore affinity after importing torch
-Process().cpu_affinity(affinity)
+if affinity is not None:
+    # Restore affinity after importing torch
+    Process().cpu_affinity(affinity)
 import os
 
 if torch.cuda.is_available():