feature: new rng engines support (#2228)

* new engines support * minor fix
uxlfoundation · Dec 19, 2024 · d9cd092 · d9cd092
1 parent 850b86d
commit d9cd092
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/generator/wrappers.py b/generator/wrappers.py
@@ -106,6 +106,8 @@ def wrap_algo(algo, ver):
     "algorithms::engines::mt19937::Batch": {"seed": ["size_t", "seed"]},
     "algorithms::engines::mt2203::Batch": {"seed": ["size_t", "seed"]},
     "algorithms::engines::mcg59::Batch": {"seed": ["size_t", "seed"]},
+    "algorithms::engines::mrg32k3a::Batch": {"seed": ["size_t", "seed"]},
+    "algorithms::engines::philox4x32x10::Batch": {"seed": ["size_t", "seed"]},
 }
 
 # Some algorithms require a setup function, to provide input without actual compute
@@ -885,6 +887,12 @@ def wrap_algo(algo, ver):
     "algorithms::engines::mt2203": [
         "ParameterType",
     ],
+    "algorithms::engines::mrg32k3a": [
+        "ParameterType",
+    ],
+    "algorithms::engines::philox4x32x10": [
+        "ParameterType",
+    ],
     "algorithms::gbt": [
         "Result",
     ],