chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,81 @@
|
||||
// Copyright (c) 2023 PaddlePaddle Authors. All Rights Reserved.
|
||||
//
|
||||
// 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 <memory>
|
||||
#include <set>
|
||||
#include <type_traits>
|
||||
|
||||
#include "paddle/pir/include/core/type_id.h"
|
||||
#include "paddle/pir/include/core/utils.h"
|
||||
|
||||
namespace pir {
|
||||
|
||||
class IR_API InterfaceValue {
|
||||
public:
|
||||
template <typename Interface, typename Model>
|
||||
static InterfaceValue Get();
|
||||
TypeId type_id() const { return type_id_; }
|
||||
void *model() const { return model_.get(); }
|
||||
|
||||
InterfaceValue() = default;
|
||||
InterfaceValue(TypeId type_id) : type_id_(type_id) {} // NOLINT
|
||||
InterfaceValue(const InterfaceValue &) = delete;
|
||||
InterfaceValue(InterfaceValue &&) noexcept;
|
||||
InterfaceValue &operator=(const InterfaceValue &) = delete;
|
||||
InterfaceValue &operator=(InterfaceValue &&) noexcept;
|
||||
~InterfaceValue() = default;
|
||||
void swap(InterfaceValue &&val) {
|
||||
using std::swap;
|
||||
swap(type_id_, val.type_id_);
|
||||
swap(model_, val.model_);
|
||||
}
|
||||
|
||||
///
|
||||
/// \brief Comparison operations.
|
||||
///
|
||||
inline bool operator<(const InterfaceValue &other) const {
|
||||
return type_id_ < other.type_id_;
|
||||
}
|
||||
|
||||
private:
|
||||
TypeId type_id_;
|
||||
std::unique_ptr<void, decltype(static_cast<void (*)(void *)>(free))> model_{
|
||||
nullptr, static_cast<void (*)(void *)>(free)};
|
||||
};
|
||||
|
||||
template <typename Interface, typename Model>
|
||||
InterfaceValue InterfaceValue::Get() {
|
||||
InterfaceValue val;
|
||||
val.type_id_ = TypeId::get<Interface>();
|
||||
static_assert(std::is_base_of<typename Interface::Concept, Model>::value,
|
||||
"Model must derived from corresponding Interface Concept.");
|
||||
static_assert(
|
||||
sizeof(typename Interface::Concept) == sizeof(Model),
|
||||
"Compared with Concept, Model class shouldn't define new data members");
|
||||
|
||||
void *model_raw = malloc(sizeof(Model));
|
||||
if (model_raw == nullptr) {
|
||||
throw("Alloc memory for interface failed.");
|
||||
}
|
||||
static_assert(std::is_trivially_destructible<Model>::value,
|
||||
"interface models must be trivially destructible");
|
||||
new (model_raw) Model();
|
||||
val.model_.reset(model_raw);
|
||||
return val;
|
||||
}
|
||||
|
||||
using InterfaceSet = std::set<InterfaceValue>;
|
||||
} // namespace pir
|
||||
Reference in New Issue
Block a user