Bugfix: Set return type of torch_tensor_get_sizes to be long long int*.

Cambridge-ICCS · Oct 22, 2024 · 5515f9e · 5515f9e
1 parent 8121ec6
commit 5515f9e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ctorch.cpp b/src/ctorch.cpp
@@ -234,7 +234,7 @@ int torch_tensor_get_rank(const torch_tensor_t tensor)
   return t->sizes().size();
 }
 
-const long int* torch_tensor_get_sizes(const torch_tensor_t tensor)
+const long long int* torch_tensor_get_sizes(const torch_tensor_t tensor)
 {
   auto t = reinterpret_cast<torch::Tensor*>(tensor);
   return t->sizes().data();

diff --git a/src/ctorch.h b/src/ctorch.h
@@ -125,7 +125,7 @@ EXPORT_C int torch_tensor_get_rank(const torch_tensor_t tensor);
  * @param Torch Tensor to determine the rank of
  * @return pointer to the sizes array of the Torch Tensor
  */
-EXPORT_C const long int* torch_tensor_get_sizes(const torch_tensor_t tensor);
+EXPORT_C const long long int* torch_tensor_get_sizes(const torch_tensor_t tensor);
 
 /**
  * Function to delete a Torch Tensor to clean up