wip

Cambridge-ICCS · Nov 15, 2024 · 7474b89 · 7474b89
1 parent eacbebc
commit 7474b89
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/ctorch.h b/src/ctorch.h
@@ -125,7 +125,11 @@ 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
  */
+#ifdef UNIX
 EXPORT_C const long int* torch_tensor_get_sizes(const torch_tensor_t tensor);
+#else
+EXPORT_C const long long int* torch_tensor_get_sizes(const torch_tensor_t tensor);
+#endif
 
 /**
  * Function to delete a Torch Tensor to clean up