Warning fixes in DLpack integration on Win32.

This commit is contained in:
Alexander Smorkalov 2025-10-02 09:12:05 +03:00
parent e9bded6ff3
commit 80fef356ec

View File

@ -112,7 +112,7 @@ bool parseDLPackTensor(DLManagedTensor* tensor, cv::cuda::GpuMat& obj, bool copy
static_cast<int>(tensor->dl_tensor.shape[1]), static_cast<int>(tensor->dl_tensor.shape[1]),
type, type,
tensor->dl_tensor.data, tensor->dl_tensor.data,
tensor->dl_tensor.strides[0] * tensor->dl_tensor.dtype.bits / 8 static_cast<size_t>(tensor->dl_tensor.strides[0] * tensor->dl_tensor.dtype.bits / 8)
); );
if (copy) if (copy)
obj = obj.clone(); obj = obj.clone();
@ -140,7 +140,7 @@ bool parseDLPackTensor(DLManagedTensor* tensor, cv::cuda::GpuMatND& obj, bool co
std::vector<int> sizes(tensor->dl_tensor.ndim); std::vector<int> sizes(tensor->dl_tensor.ndim);
for (int i = 0; i < tensor->dl_tensor.ndim - 1; ++i) for (int i = 0; i < tensor->dl_tensor.ndim - 1; ++i)
{ {
steps[i] = tensor->dl_tensor.strides[i] * tensor->dl_tensor.dtype.bits / 8; steps[i] = static_cast<size_t>(tensor->dl_tensor.strides[i] * tensor->dl_tensor.dtype.bits / 8);
sizes[i] = static_cast<int>(tensor->dl_tensor.shape[i]); sizes[i] = static_cast<int>(tensor->dl_tensor.shape[i]);
} }
sizes.back() = static_cast<int>(tensor->dl_tensor.shape[tensor->dl_tensor.ndim - 1]); sizes.back() = static_cast<int>(tensor->dl_tensor.shape[tensor->dl_tensor.ndim - 1]);