Skip to content

Commit d9cd092

Browse files
feature: new rng engines support (uxlfoundation#2228)
* new engines support * minor fix
1 parent 850b86d commit d9cd092

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

generator/wrappers.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,6 +106,8 @@ def wrap_algo(algo, ver):
106106
"algorithms::engines::mt19937::Batch": {"seed": ["size_t", "seed"]},
107107
"algorithms::engines::mt2203::Batch": {"seed": ["size_t", "seed"]},
108108
"algorithms::engines::mcg59::Batch": {"seed": ["size_t", "seed"]},
109+
"algorithms::engines::mrg32k3a::Batch": {"seed": ["size_t", "seed"]},
110+
"algorithms::engines::philox4x32x10::Batch": {"seed": ["size_t", "seed"]},
109111
}
110112

111113
# Some algorithms require a setup function, to provide input without actual compute
@@ -885,6 +887,12 @@ def wrap_algo(algo, ver):
885887
"algorithms::engines::mt2203": [
886888
"ParameterType",
887889
],
890+
"algorithms::engines::mrg32k3a": [
891+
"ParameterType",
892+
],
893+
"algorithms::engines::philox4x32x10": [
894+
"ParameterType",
895+
],
888896
"algorithms::gbt": [
889897
"Result",
890898
],

0 commit comments

Comments
 (0)