diff --git a/torchcsprng/csrc/kernels_commons.h b/torchcsprng/csrc/kernels_commons.h index c94167e..3e74d35 100644 --- a/torchcsprng/csrc/kernels_commons.h +++ b/torchcsprng/csrc/kernels_commons.h @@ -39,6 +39,8 @@ struct CSPRNGGeneratorImpl : public c10::GeneratorImpl { void set_state(const c10::TensorImpl& new_state) override { throw std::runtime_error("not implemented"); } c10::intrusive_ptr get_state() const override { throw std::runtime_error("not implemented"); } + void set_offset(uint64_t offset) override { throw std::runtime_error("not implemented"); } + uint64_t get_offset() const override { throw std::runtime_error("not implenented"); } bool use_rd_; std::random_device rd_; std::mt19937 mt_;