from elementSizeInBytes to element_size, following upstream commit ht… (

horovod#919)

* from elementSizeInBytes to element_size, following upstream commit pytorch/pytorch#17785

Signed-off-by: labor00 <abrvb@outlook.com>

* uses TORCH_VERSION macro to ensure backward compatibility

Signed-off-by: labor00 <abrvb@outlook.com>

* if pytorch version string contains dev return a very big number

Signed-off-by: labor00 <abrvb@outlook.com>
Signed-off-by: Yana Shchyokotova <yana.shchyokotova@intel.com>

horovod/torch/adapter_v2.cc

@@ -70,7 +70,11 @@ const TensorShape TorchTensor::shape() const {
 const void* TorchTensor::data() const { return tensor_.data_ptr(); }
 
 int64_t TorchTensor::size() const {
+# if TORCH_VERSION >= 1001000000
+  return tensor_.element_size() * tensor_.numel();
+#else
   return tensor_.type().elementSizeInBytes() * tensor_.numel();
+#endif
 }
 
 TorchOpContext::TorchOpContext(int device, ::torch::Tensor output)

setup.py

@@ -623,6 +623,8 @@ def build_tf_extension(build_ext, options):
 
 
 def parse_version(version_str):
+    if "dev" in version_str:
+        return 9999999999
     m = re.match('^(\d+)(?:\.(\d+))?(?:\.(\d+))?(?:\.(\d+))?', version_str)
     if m is None:
         return None