pinned_memory.h Source File

pinned_memory.h Source File#

SDK qb Runtime Library: pinned_memory.h Source File

SDK qb Runtime Library v1.3

MCS001-EN

// Copyright ⓒ 2019- Mobilint Inc. All rights reserved.
 
#ifndef QBRUNTIME_PINNED_MEMORY_H_
#define QBRUNTIME_PINNED_MEMORY_H_
 
#include <stdint.h>
 
#include <memory>
#include <vector>
 
#include "qbruntime/export.h"
 
namespace mobilint {
 
class PinnedMemoryImpl;
template <typename T>
class QBRUNTIME_EXPORT PinnedMemory {
public:
    PinnedMemory() = delete;
    ~PinnedMemory();
    PinnedMemory(const PinnedMemory& other);
    PinnedMemory& operator=(const PinnedMemory& other);
    PinnedMemory(PinnedMemory&& other) noexcept;
    PinnedMemory& operator=(PinnedMemory&& other) noexcept;
    T* data() const;
    const std::vector<int64_t>& shape() const;
    uint64_t size() const;
    bool isAvailable() const;
    PinnedMemory(const std::vector<int64_t>& shape, uint64_t phys_addr, void* mapped_ptr);
 
private:
    PinnedMemory(std::shared_ptr<PinnedMemoryImpl> pinned_memory_impl);
 
    std::shared_ptr<PinnedMemoryImpl> mImpl;
 
    friend class PinnedMemoryImpl;
    friend class ModelImpl;
    friend class AcceleratorImpl;
};
 
}  // namespace mobilint
 
#endif  // QBRUNTIME_PINNED_MEMORY_H_