Add getDict function to PyBinds

Given a CudaMappingOptions instance, return a Python dictionary
that represents all the given options and their values.

Author: Jules Pondard

tensor_comprehensions/pybinds/tclib.cc

@@ -654,6 +654,38 @@ PYBIND11_MODULE(tclib, m) {
             return str;
           },
           "Returns the CudaMappingOptions as a human-readable string")
+      .def(
+          "getDict",
+          [](tc::CudaMappingOptions& instance) {
+            py::dict rv;
+            rv["outerScheduleFusionStrategy"] = FusionStrategy_Name(
+                instance.generic.outerScheduleOptions.proto.fusion_strategy());
+            if (instance.generic.proto.has_intra_tile_schedule_options())
+              rv["intraTileScheduleFusionStrategy"] =
+                  FusionStrategy_Name(instance.generic.intraTileScheduleOptions
+                                          .proto.fusion_strategy());
+            rv["fixParametersBeforeScheduling"] =
+                instance.generic.proto.fix_parameters_before_scheduling();
+            if (instance.generic.proto.has_tiling())
+              rv["tile"] = instance.generic.tiling.extractVector();
+            if (instance.generic.proto.has_unroll())
+              rv["unroll"] = instance.generic.proto.unroll();
+            rv["tileImperfectlyNested"] =
+                instance.generic.proto.tile_imperfectly_nested();
+            rv["matchLibraryCalls"] =
+                instance.generic.proto.match_library_calls();
+            rv["mapToThreads"] = instance.block.extractVector();
+            rv["mapToBlocks"] = instance.grid.extractVector();
+            rv["useSharedMemory"] = instance.proto().use_shared_memory();
+            rv["usePrivateMemory"] = instance.proto().use_private_memory();
+            rv["unrollCopyShared"] = instance.proto().unroll_copy_shared();
+            rv["useReadOnlyCache"] = instance.proto().use_readonly_cache();
+            if (instance.proto().has_max_shared_memory())
+              rv["maxSharedMemory"] = instance.proto().max_shared_memory();
+            rv["privateDepth"] = instance.proto().private_depth();
+            return rv;
+          },
+          "Returns a dictionary with the CudaMappingOptions")
       .def(
           "serialize",
           [](tc::CudaMappingOptions& instance) {