Use custom mapping for serializing ATen types

ebetica / autogradpp

Direct C++ Interface to PyTorch

MIT License

80 stars 12 forks source link

Use custom mapping for serializing ATen types #66

Closed jgehring closed 6 years ago

jgehring commented 6 years ago

The autogradpp serialization should be robust to ATen changes like https://git.io/vxd6R (likely due to clang-format). Hence, let's use custom mappings to store at::Backend and at::ScalarType as integers.