[ez] Make NUMA signpost parameters JSON serializable (pytorch#160710)

# Context
Broader context in pytorch#160163.

In order for the _utils_internal version of signpost_event to do proper logging, its parameters argument needs to be json serializable.

# This PR
Convert `NumaOptions` to serializable form before inputting to `signpost_event`.

# Test Plan
## Automated
Added tests `$ pytest test/test_numa_binding.py`.

## Manual
See [D80317206](https://www.internalfb.com/diff/D80317206).

Pull Request resolved: pytorch#160710
Approved by: https://github.com/kiukchung

Author: pdesupinski

test/test_numa_binding.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import json
 import multiprocessing.spawn as spawn
 import os
 import subprocess
@@ -13,6 +14,7 @@
 from unittest.mock import mock_open, patch
 
 import torch
+from torch._utils_internal import signpost_event
 from torch.distributed.elastic.multiprocessing import DefaultLogsSpecs, start_processes
 from torch.numa.binding import (
     _get_ranges_str_from_ints,
@@ -68,6 +70,7 @@ def setUp(self) -> None:
             patch("shutil.which", return_value="/usr/bin/numactl"),
             patch("torch.numa.binding.run"),
             patch("torch.numa.binding.mkstemp", self._mock_mkstemp),
+            patch("torch.numa.binding.signpost_event", self._mock_signpost_event),
         ]
 
         for context_manager in self._context_managers_to_apply_to_all_tests:
@@ -86,6 +89,11 @@ def tearDown(self) -> None:
             context_manager.__exit__(None, None, None)
         super().tearDown()
 
+    def _mock_signpost_event(self, *args, **kwargs) -> None:
+        # Please keep these parameters JSON serializable for logging purposes
+        json.dumps(kwargs["parameters"])
+        return signpost_event(*args, **kwargs)
+
     def _mock_mkstemp(self, *args, **kwargs):
         # Just keep track of temp files so we can delete them
         fd, path = _real_mkstemp(*args, **kwargs)

torch/numa/binding.py

@@ -5,7 +5,7 @@
 import traceback
 from collections import defaultdict
 from collections.abc import Iterable
-from dataclasses import dataclass
+from dataclasses import asdict, dataclass
 from enum import Enum
 from logging import getLogger
 from subprocess import run
@@ -113,7 +113,7 @@ def maybe_wrap_command_with_numa_bindings(
     kwargs = {
         "command_args": command_args,
         "gpu_index": gpu_index,
-        "numa_options": numa_options,
+        "numa_options": asdict(numa_options),
     }
     logger.info("Attempting to wrap command with NUMA bindings, given input %r", kwargs)