frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make sure that little endian is default case when __BYTE_ORDER__ is not defined (#104249)

Browse Source 
This is a follow up to discussion
in https://github.com/pytorch/pytorch/pull/96422

Pull Request resolved: https://github.com/pytorch/pytorch/pull/104249
Approved by: https://github.com/malfet
This commit is contained in:
Aleksei Nikiforov
2023-07-27 13:33:35 +00:00
committed by PyTorch MergeBot
parent 7c97c943fb
commit 6f1042c049
4 changed files with 15 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/csrc/jit/serialization/pickler.cpp
View File

@ -552,9 +552,11 @@ void Pickler::pushDouble(double value) {
#if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
// Python pickle format is big endian, swap.
push<double>(swapDouble(value));
#else /* __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ */
#elif __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
push<double>(value);
#endif /* __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ */
#else
#error Unexpected or undefined __BYTE_ORDER__
#endif
}
void Pickler::pushComplexDouble(const IValue& value) {
c10::complex<double> d = value.toComplexDouble();

6
torch/csrc/jit/serialization/unpickler.cpp
View File

@ -230,9 +230,11 @@ double Unpickler::readFloat() {
reinterpret_cast<char*>(&little_endian));
return little_endian;
#else /* __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ */
#elif __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
return big_endian;
#endif /* __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ */
#else
#error Unexpected or undefined __BYTE_ORDER__
#endif
}
void Unpickler::run() {

6
torch/csrc/jit/tensorexpr/tensorexpr_init.cpp
View File

@ -836,7 +836,7 @@ void initTensorExprBindings(PyObject* module) {
value_ptrs.emplace_back(value.cast<at::Tensor>().data_ptr());
}
}
#else /* __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ */
#elif __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
if (py::len(values) != self.buffer_args().size()) {
throw malformed_input("bad args in CodeGen.call function");
}
@ -862,7 +862,9 @@ void initTensorExprBindings(PyObject* module) {
value_ptrs.emplace_back(value.cast<at::Tensor>().data_ptr());
}
}
#endif /* __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__ */
#else
#error Unexpected or undefined __BYTE_ORDER__
#endif
self.call(value_ptrs);
})
.def(

4
torch/csrc/utils/byte_order.h
View File

@ -43,7 +43,7 @@
#define from_le32(x) (x)
#define to_le64(x) (x)
#define from_le64(x) (x)
#else
#elif __BYTE_ORDER__ == __ORDER_BIG_ENDIAN__
#define to_be16(x) (x)
#define from_be16(x) (x)
#define to_be32(x) (x)
@ -56,6 +56,8 @@
#define from_le32(x) thp_bswap32(x)
#define to_le64(x) thp_bswap64(x)
#define from_le64(x) thp_bswap64(x)
#else
#error Unexpected or undefined __BYTE_ORDER__
#endif
namespace torch {