Function CreateGenerator

oneflow/api/python/framework/random_generator.cpp:29–32

namespace oneflow {

Maybe<one::Generator> CreateGenerator(const std::string& device_str) {
  auto [device_name, device_index, rematable] = *JUST(ParseDeviceString(device_str));
  return one::MakeGenerator(device_name, device_index);
}

py::tuple GetCudaDefaultGenerators() {
#ifdef WITH_CUDA