@@ -54,4 +54,10 @@ class Index : public Uniform<size_t> {
public:
explicit Index(size_t size) : Uniform(0, size - 1) {}
};
+
+class Boolean : public Uniform<int> {
+public:
+ Boolean() : Uniform(0, 1) {}
+ void describe(std::ostream &os) const override;
+};
}
@@ -46,5 +46,7 @@ template <typename Rand> void Uniform<Rand>::describe(std::ostream & os) const {
os << typeid(Rand).name() << "Uniform[" << min_ << ',' << max_ << close;
+void Boolean::describe(std::ostream &os) const { os << "Boolean"; }
IMPLEMENT_DEVICE(INSTANTIATE_DISTRIBUTION_TEMPLATES)