Program Listing for File SharedLibLoader.hpp

↰ Return to documentation for file (pennylane_lightning/core/src/utils/SharedLibLoader.hpp)

// Copyright 2018-2024 Xanadu Quantum Technologies Inc.

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at

//     http://www.apache.org/licenses/LICENSE-2.0

// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#pragma once
#include <string>

#if defined(__APPLE__) || defined(__linux__)
#include <dlfcn.h>
#define PL_DLOPEN(NAME, ARG) dlopen(NAME, ARG)
#define PL_DLERROR() dlerror()
#define PL_DLCLOSE(NAME) dlclose(NAME)
#define PL_DLSYS(NAME, SYMBOL) dlsym(NAME, SYMBOL)
#endif

#include "Error.hpp"

namespace Pennylane::Util {
// Ignore invalid warnings for compile-time checks
// NOLINTBEGIN
class SharedLibLoader final {
  private:
    void *handle_{nullptr};

  public:
    SharedLibLoader();
    explicit SharedLibLoader(const std::string &filename) {
        handle_ = PL_DLOPEN(filename.c_str(), RTLD_LAZY);
        PL_ABORT_IF(!handle_, PL_DLERROR());
    }

    ~SharedLibLoader() noexcept { PL_DLCLOSE(handle_); }

    void *getHandle() { return handle_; }

    void *getSymbol(const std::string &symbol) {
        void *sym = PL_DLSYS(handle_, symbol.c_str());
        PL_ABORT_IF(!sym, PL_DLERROR());
        return sym;
    }
};
// NOLINTEND

} // namespace Pennylane::Util

api/program_listing_file_pennylane_lightning_core_src_utils_SharedLibLoader.hpp

 

Download Python script

 

Download Notebook

 

View on GitHub