feat: support Brownian tree noise in all noise injection samplers (#1899)

This commit is contained in:
Wagner Bruna authored and GitHub committed 2026-09-15 02:37:31 +08:00
1 parent f9ddc0f388
commit 07a85c74cb
5 files changed
+233 -191

No files matched your search

+6
View File
@@ -1,7 +1,9 @@
#ifndef __SD_CORE_RNG_MT19937_HPP__
#define __SD_CORE_RNG_MT19937_HPP__
#include <array>
#include <cmath>
#include <limits>
#include <vector>
#include "core/rng.hpp"
@@ -123,6 +125,10 @@ class MT19937RNG : public RNG {
public:
MT19937RNG(uint64_t seed = 0) { manual_seed(seed); }
std::shared_ptr<RNG> clone() const override {
return std::make_shared<MT19937RNG>(*this);
}
void manual_seed(uint64_t seed) override {
s.seed_ = seed;
s.seeded_ = true;