Skip to content

Commit 23cdf2d

Browse files
committed
fix build
1 parent 4af1315 commit 23cdf2d

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

onnxruntime/contrib_ops/cpu/moe/moe_quantization_cpu.cc

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -598,8 +598,8 @@ Status QMoECPU<T>::UseSharedPrePackedBuffers(std::vector<BufferUniquePtr>& prepa
598598
if (prepacked_buffers.size() > 1) {
599599
int64_t* buffer_data = static_cast<int64_t*>(prepacked_buffers[1].get());
600600
int64_t rank = buffer_data[0];
601-
std::vector<int64_t> dims(rank);
602-
memcpy(dims.data(), buffer_data + 1, rank * sizeof(int64_t));
601+
std::vector<int64_t> dims(static_cast<size_t>(rank));
602+
memcpy(dims.data(), buffer_data + 1, static_cast<size_t>(rank) * sizeof(int64_t));
603603
fc1_shape_ = TensorShape(dims);
604604
}
605605
if (prepacked_buffers.size() > 2) {
@@ -611,8 +611,8 @@ Status QMoECPU<T>::UseSharedPrePackedBuffers(std::vector<BufferUniquePtr>& prepa
611611
if (prepacked_buffers.size() > 1) {
612612
int64_t* buffer_data = static_cast<int64_t*>(prepacked_buffers[1].get());
613613
int64_t rank = buffer_data[0];
614-
std::vector<int64_t> dims(rank);
615-
memcpy(dims.data(), buffer_data + 1, rank * sizeof(int64_t));
614+
std::vector<int64_t> dims(static_cast<size_t>(rank));
615+
memcpy(dims.data(), buffer_data + 1, static_cast<size_t>(rank) * sizeof(int64_t));
616616
fc2_shape_ = TensorShape(dims);
617617
}
618618
if (prepacked_buffers.size() > 2) {
@@ -624,8 +624,8 @@ Status QMoECPU<T>::UseSharedPrePackedBuffers(std::vector<BufferUniquePtr>& prepa
624624
if (prepacked_buffers.size() > 1) {
625625
int64_t* buffer_data = static_cast<int64_t*>(prepacked_buffers[1].get());
626626
int64_t rank = buffer_data[0];
627-
std::vector<int64_t> dims(rank);
628-
memcpy(dims.data(), buffer_data + 1, rank * sizeof(int64_t));
627+
std::vector<int64_t> dims(static_cast<size_t>(rank));
628+
memcpy(dims.data(), buffer_data + 1, static_cast<size_t>(rank) * sizeof(int64_t));
629629
fc3_shape_ = TensorShape(dims);
630630
}
631631
used_shared_buffers = true;

0 commit comments

Comments
 (0)