mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
483f748dd5
This PR enables `-Winconsistent-missing-destructor-override` and `-Winconsistent-missing-override` and fixes violations. <!-- copilot:summary --> ### <samp>🤖 Generated by Copilot at 47e904e</samp> This pull request updates the code of various classes and operators in the `caffe2` and `aten` subdirectories to use the `override` specifier instead of the `virtual` keyword for destructors and other virtual functions that override a base class function. This improves the code readability, quality, and consistency with C++ best practices. It also modifies the `./CMakeLists.txt` file to enable warnings for these specifiers, but disable errors. Pull Request resolved: https://github.com/pytorch/pytorch/pull/104032 Approved by: https://github.com/malfet
89 lines
2.6 KiB
C++
89 lines
2.6 KiB
C++
#ifndef CAFFE2_CORE_QTENSOR_SERIALIZATION_H_
|
|
#define CAFFE2_CORE_QTENSOR_SERIALIZATION_H_
|
|
|
|
#include "caffe2/core/blob_serialization.h"
|
|
#include "caffe2/core/qtensor.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
constexpr auto kQTensorBlobQType = "QTensor";
|
|
|
|
template <class Context>
|
|
class QTensorSerializer : public BlobSerializerBase {
|
|
public:
|
|
QTensorSerializer() : context_() {}
|
|
~QTensorSerializer() override {}
|
|
/**
|
|
* Serializes a Blob. Note that this blob has to contain QTensor<Context>.
|
|
*/
|
|
void Serialize(
|
|
const void* pointer,
|
|
TypeMeta typeMeta,
|
|
const string& name,
|
|
SerializationAcceptor acceptor) override;
|
|
|
|
private:
|
|
Context context_;
|
|
};
|
|
|
|
template <class Context>
|
|
class QTensorDeserializer : public BlobDeserializerBase {
|
|
public:
|
|
void Deserialize(const BlobProto& proto, Blob* blob) override;
|
|
void Deserialize(const QTensorProto& proto, QTensor<Context>* tensor);
|
|
};
|
|
|
|
template <class Context>
|
|
void QTensorSerializer<Context>::Serialize(
|
|
const void* pointer,
|
|
TypeMeta typeMeta,
|
|
const string& name,
|
|
BlobSerializerBase::SerializationAcceptor acceptor) {
|
|
CAFFE_ENFORCE(typeMeta.Match<QTensor<Context>>());
|
|
const auto& qtensor = *static_cast<const QTensor<Context>*>(pointer);
|
|
BlobProto blob_proto;
|
|
blob_proto.set_name(name);
|
|
blob_proto.set_type(kQTensorBlobQType);
|
|
QTensorProto& proto = *blob_proto.mutable_qtensor();
|
|
proto.set_name(name);
|
|
for (const auto i : c10::irange(qtensor.ndim())) {
|
|
proto.add_dims(qtensor.dim32(i));
|
|
}
|
|
proto.set_precision(qtensor.precision());
|
|
proto.set_scale(qtensor.scale());
|
|
proto.set_bias(qtensor.bias());
|
|
proto.set_is_signed(qtensor.is_signed());
|
|
detail::CopyToProtoWithCast(
|
|
qtensor.nbytes(), qtensor.data(), proto.mutable_data(), &this->context_);
|
|
acceptor(name, SerializeBlobProtoAsString_EnforceCheck(blob_proto));
|
|
}
|
|
|
|
template <class Context>
|
|
void QTensorDeserializer<Context>::Deserialize(
|
|
const BlobProto& blob_proto,
|
|
Blob* blob) {
|
|
Deserialize(blob_proto.qtensor(), blob->GetMutable<QTensor<Context>>());
|
|
}
|
|
|
|
template <class Context>
|
|
void QTensorDeserializer<Context>::Deserialize(
|
|
const QTensorProto& proto,
|
|
QTensor<Context>* qtensor) {
|
|
Context context{};
|
|
vector<int> dims;
|
|
for (const int d : proto.dims()) {
|
|
dims.push_back(d);
|
|
}
|
|
qtensor->Resize(dims);
|
|
qtensor->SetPrecision(proto.precision());
|
|
qtensor->SetScale(proto.scale());
|
|
qtensor->SetBias(proto.bias());
|
|
qtensor->SetSigned(proto.is_signed());
|
|
|
|
detail::CopyFromProtoWithCast(
|
|
qtensor->nbytes(), proto.data(), qtensor->mutable_data(), &context);
|
|
}
|
|
|
|
} // namespace caffe2
|
|
|
|
#endif // CAFFE2_CORE_QTENSOR_SERIALIZATION_H_
|