File size: 679 Bytes
2216aae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 | #pragma once
#include <ATen/Utils.h>
#include <c10/macros/Export.h>
#include <c10/util/Exception.h>
namespace c10 {
class DynamicLibraryError : public Error {
using Error::Error;
};
} // namespace c10
namespace at {
struct DynamicLibrary {
AT_DISALLOW_COPY_AND_ASSIGN(DynamicLibrary);
DynamicLibrary(DynamicLibrary&& other) = delete;
DynamicLibrary& operator=(DynamicLibrary&&) = delete;
TORCH_API DynamicLibrary(
const char* name,
const char* alt_name = nullptr,
bool leak_handle = false);
TORCH_API void* sym(const char* name);
TORCH_API ~DynamicLibrary();
private:
bool leak_handle;
void* handle = nullptr;
};
} // namespace at
|