diff --git a/src/device_image.h b/src/device_image.h index 71566e3..f7c5b70 100644 --- a/src/device_image.h +++ b/src/device_image.h @@ -45,8 +45,8 @@ class DeviceImage void download(void* data) const; void fill_zero(); - template T* ptr(int y = 0) { return (T*)data + y * step; } - template const T* ptr(int y = 0) const { return (T*)data + y * step; } + template T* ptr(int y = 0) { return (T*)data + y * (size_t)step; } + template const T* ptr(int y = 0) const { return (T*)data + y * (size_t)step; } void* data; int rows, cols, step;