@@ -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