You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
81 lines
2.5 KiB
81 lines
2.5 KiB
#if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|
|
//===- llvm/ADT/STLExtras.h - Useful STL related functions ------*- C++ -*-===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file contains some templates that are useful if you are working with the
|
|
// STL at all.
|
|
//
|
|
// No library is required when using these functions.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// c10: modified from llvm::function_ref
|
|
// c10: added more SFINAE to enable use in overloaded functions
|
|
|
|
#pragma once
|
|
|
|
#include <cstdint>
|
|
#include <type_traits>
|
|
#include <utility>
|
|
|
|
namespace c10 {
|
|
|
|
/// An efficient, type-erasing, non-owning reference to a callable. This is
|
|
/// intended for use as the type of a function parameter that is not used
|
|
/// after the function in question returns.
|
|
///
|
|
/// This class does not own the callable, so it is not in general safe to store
|
|
/// a function_ref.
|
|
template <typename Fn>
|
|
class function_ref;
|
|
|
|
template <typename Ret, typename... Params>
|
|
class function_ref<Ret(Params...)> {
|
|
Ret (*callback)(intptr_t callable, Params... params) = nullptr;
|
|
intptr_t callable{};
|
|
|
|
template <typename Callable>
|
|
static Ret callback_fn(intptr_t callable, Params... params) {
|
|
return (*reinterpret_cast<Callable*>(callable))(
|
|
std::forward<Params>(params)...);
|
|
}
|
|
|
|
public:
|
|
function_ref() = default;
|
|
function_ref(std::nullptr_t) {}
|
|
|
|
template <typename Callable>
|
|
function_ref(
|
|
// NOLINTNEXTLINE(cppcoreguidelines-missing-std-forward)
|
|
Callable&& callable,
|
|
std::enable_if_t<!std::is_same_v<
|
|
std::remove_reference_t<Callable>,
|
|
function_ref>>* /*unused*/
|
|
= nullptr,
|
|
std::enable_if_t<std::is_convertible_v<
|
|
typename std::invoke_result_t<Callable, Params...>,
|
|
Ret>>* /*unused*/
|
|
= nullptr)
|
|
: callback(callback_fn<std::remove_reference_t<Callable>>),
|
|
callable(reinterpret_cast<intptr_t>(&callable)) {}
|
|
|
|
Ret operator()(Params... params) const {
|
|
return callback(callable, std::forward<Params>(params)...);
|
|
}
|
|
|
|
operator bool() const {
|
|
return callback;
|
|
}
|
|
};
|
|
|
|
} // namespace c10
|
|
|
|
#else
|
|
#error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
|
|
#endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|