Skip to content

Commit 407a92d

Browse files
kashifapaszke
authored andcommitted
std::min() requires same type (pytorch#732)
* std::min() requires same type * cast buffer instead * declare buffer_size as int64_t
1 parent 0a893ab commit 407a92d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torch/csrc/generic/serialization.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ void THPStorage_(writeFileRaw)(THStorage *self, int fd)
4949
remaining -= result;
5050
}
5151
} else {
52-
long buffer_size = std::min(self->size, (long)5000);
52+
int64_t buffer_size = std::min(self->size, (long)5000);
5353
std::unique_ptr<uint8_t[]> le_buffer(new uint8_t[buffer_size * sizeof(real)]);
5454
for (int64_t i = 0; i < self->size; i += buffer_size) {
5555
size_t to_convert = std::min(self->size - i, buffer_size);
@@ -100,7 +100,7 @@ THStorage * THPStorage_(readFileRaw)(int fd)
100100
remaining -= result;
101101
}
102102
} else {
103-
long buffer_size = std::min(size, (long)5000);
103+
int64_t buffer_size = std::min(size, (long)5000);
104104
std::unique_ptr<uint8_t[]> le_buffer(new uint8_t[buffer_size * sizeof(real)]);
105105
for (int64_t i = 0; i < size; i += buffer_size) {
106106
size_t to_convert = std::min(size - i, buffer_size);

0 commit comments

Comments
 (0)