Llama-3.1-8B-DALv0.1
/
venv
/lib
/python3.12
/site-packages
/torch
/include
/ATen
/detail
/MAIAHooksInterface.h
// NB: Class must live in `at` due to limitations of Registry.h. | |
namespace at { | |
struct TORCH_API MAIAHooksInterface { | |
// This should never actually be implemented, but it is used to | |
// squelch -Werror=non-virtual-dtor | |
virtual ~MAIAHooksInterface() = default; | |
virtual std::string showConfig() const { | |
TORCH_CHECK(false, "Cannot query detailed MAIA version information."); | |
} | |
}; | |
// NB: dummy argument to suppress "ISO C++11 requires at least one argument | |
// for the "..." in a variadic macro" | |
struct TORCH_API MAIAHooksArgs {}; | |
TORCH_DECLARE_REGISTRY(MAIAHooksRegistry, MAIAHooksInterface, MAIAHooksArgs); | |
namespace detail { | |
TORCH_API const MAIAHooksInterface& getMAIAHooks(); | |
} // namespace detail | |
} // namespace at | |