Function PYBIND11_MODULE

dali/python/backend_impl.cc:3084–3615

}

PYBIND11_MODULE(backend_impl, m, py::mod_gil_not_used()) {
  dali::InitOperatorsLib();
  m.doc() = "Python bindings for the C++ portions of DALI";

  // DALI Init function
  m.def("Init", &DALIInit);

  ExposeBufferPolicyFunctions(m);

  m.def("LoadLibrary", &PluginManager::LoadLibrary,
    py::arg("lib_path"),
    py::arg("global_symbols") = false,
    py::arg("allow_fail") = false);

  m.def("LoadDirectory", &PluginManager::LoadDirectory,
    py::arg("dir_path"),
    py::arg("global_symbols") = false,
    py::arg("allow_fail") = false);

  m.def("LoadDefaultPlugins", &PluginManager::LoadDefaultPlugins);

  m.def("GetCxx11AbiFlag", &GetCxx11AbiFlag);

  m.def("IsDriverInitialized", [] {
    // we just want to check if cuda has been loaded already
    if (dlopen("libcuda.so", RTLD_NOLOAD | RTLD_NOW) ||
        dlopen("libcuda.so.1", RTLD_NOLOAD | RTLD_NOW)) {
      int place_holder = -1;
      // call cuDeviceGetCount only if cuda is loaded, if not there is not point in calling
      // a check that would load it
      return CUDA_SUCCESS == cuDeviceGetCount(&place_holder);
    }
    return false;
  });

  m.def("GetCudaVersion", [] {
    int version = -1;
    auto ret = cudaDriverGetVersion(&version);
    if (ret != cudaSuccess) {
      return -1;
    } else {
      return version;
    }
  });

  m.def("GetCufftVersion", [] {
    int ret = -1;
    try {
      // we don't want to throw when it is not available, just return -1
      ret = GetCufftVersion();
    } catch (const std::runtime_error &) {}
    return ret;
  });

  m.def("GetNppVersion", [] {
    int ret = -1;
    try {
      // we don't want to throw when it is not available, just return -1