mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
47c70a43b4
(2) blob serialization comments
(3) cudnn: putting it under a separate device name
so we can explicitly choose cudnn instead of
having CUDA device prioritizing it.
(4) note that mint is not available with ipython
due to zeromq conflict
(5) db_throughput utility
(6) added gprofiler
18 lines
588 B
C++
18 lines
588 B
C++
#include "caffe2/core/blob.h"
|
|
#include "caffe2/core/blob_serialization.h"
|
|
|
|
namespace caffe2 {
|
|
DEFINE_TYPED_REGISTRY(BlobSerializerRegistry, internal::TypeId,
|
|
BlobSerializerBase);
|
|
|
|
namespace {
|
|
REGISTER_BLOB_SERIALIZER(float_cpu,
|
|
(internal::GetTypeId<Tensor<float, CPUContext> >()),
|
|
TensorSerializerFloat<CPUContext>);
|
|
REGISTER_BLOB_SERIALIZER(int32_cpu,
|
|
(internal::GetTypeId<Tensor<int, CPUContext> >()),
|
|
TensorSerializerInt32<CPUContext>);
|
|
}
|
|
} // namespace caffe2
|
|
|