mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
29d759948e
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66746 Modified loops in files under fbsource/fbcode/caffe2/ from the format `for(TYPE var=x0;var<x_max;x++)` to the format `for(const auto var: irange(xmax))` This was achieved by running r-barnes's loop upgrader script (D28874212) with some modification to exclude all files under /torch/jit and a number of reversions or unused variable suppression warnings added by hand. Test Plan: Sandcastle Reviewed By: malfet Differential Revision: D31705361 fbshipit-source-id: 33fd22eb03086d114e2c98e56703e8ec84460268
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() {}
|
|
/**
|
|
* 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_
|