Method ThreadController

mlx/data/core/ThreadController.cpp:69–105

};

ThreadController::ThreadController() {
  std::vector<std::string> get_names = {
      "MKL_Get_Max_Threads",
      "openblas_get_num_threads",
      "openblas_get_num_threads64_",
      "omp_get_num_threads"};
  std::vector<std::string> set_names = {
      "MKL_Set_Num_Threads",
      "openblas_set_num_threads",
      "openblas_set_num_threads64_",
      "omp_set_num_threads"};

#ifdef MLX_HAS_DLOPEN_NOLOAD
  auto lib_names = find_loaded_lib_names();
  for (auto i = 0; i < get_names.size(); i++) {
    auto get_name = get_names[i];
    auto set_name = set_names[i];
    for (auto l = 0; l <= lib_names.size(); l++) {
      // at the last resort, we test main executable
      const char* lib_name =
          (l < lib_names.size() ? lib_names[l].c_str() : nullptr);
      auto lib = dlopen(lib_name, RTLD_NOW | RTLD_NOLOAD);
      if (lib) {
        auto get_sym = dlsym(lib, get_name.c_str());
        auto set_sym = dlsym(lib, set_name.c_str());
        if (get_sym && set_sym) {
          symbols_.push_back(
              std::make_shared<ThreadControllerSym>(lib, get_sym, set_sym));
          break;
        }
      } else {
        dlclose(lib);
      }
    }
  }
#endif
}

ThreadControllerState ThreadController::limit() {
  ThreadControllerState state(symbols_.size());