@@ -636,7 +636,7 @@ MemoryPtr split_horizontal(const dnnl::engine& eng,
636636 const auto & dims = shape.getDims ();
637637 auto prec = src->getPrecision ();
638638 if (dim < 0 ) {
639- dim += dims.size ();
639+ dim += static_cast < int >( dims.size () );
640640 }
641641 auto split_parts = [](int len, int n) {
642642 int average = len / n;
@@ -652,15 +652,15 @@ MemoryPtr split_horizontal(const dnnl::engine& eng,
652652 return src;
653653 }
654654 auto new_pshape = pshape;
655- auto splited_dim_vec = split_parts (new_pshape[dim].get_length (), w_size);
655+ auto splited_dim_vec = split_parts (static_cast < int >( new_pshape[dim].get_length () ), w_size);
656656 new_pshape[dim] = splited_dim_vec[w_rank];
657657
658658 auto new_desc = std::make_shared<CpuBlockedMemoryDesc>(prec, Shape{new_pshape});
659659 MemoryPtr ptr = std::make_shared<Memory>(eng, new_desc);
660660 return ptr;
661661 }
662662 assert (static_cast <int >(dims[dim]) >= w_size);
663- auto splited_dim_vec = split_parts (dims[dim], w_size);
663+ auto splited_dim_vec = split_parts (static_cast < int >( dims[dim]) , w_size);
664664
665665 // reference stride
666666 VectorDims stride_dims = dims;
@@ -700,7 +700,7 @@ MemoryPtr split_vertical(const dnnl::engine& eng,
700700 const auto & dims = shape.getDims ();
701701 auto prec = src->getPrecision ();
702702 if (dim < 0 ) {
703- dim += dims.size ();
703+ dim += static_cast < int >( dims.size () );
704704 }
705705 auto split_parts = [](int len, int n) {
706706 int average = len / n;
@@ -712,7 +712,7 @@ MemoryPtr split_vertical(const dnnl::engine& eng,
712712 const auto & pshape = shape.toPartialShape ();
713713 OPENVINO_ASSERT (!pshape[dim].is_dynamic (), " Can't split data with dynamic shapes" );
714714 auto new_pshape = pshape;
715- auto splited_dim_vec = split_parts (new_pshape[dim].get_length (), w_size);
715+ auto splited_dim_vec = split_parts (static_cast < int >( new_pshape[dim].get_length () ), w_size);
716716 new_pshape[dim] = splited_dim_vec[w_rank];
717717
718718 auto new_desc = std::make_shared<CpuBlockedMemoryDesc>(prec, Shape{new_pshape});
@@ -721,7 +721,7 @@ MemoryPtr split_vertical(const dnnl::engine& eng,
721721 }
722722 assert (static_cast <int >(dims[dim]) >= w_size);
723723 const auto splited_size = dims[dim] * prec.size ();
724- auto splited_dim_vec = split_parts (dims[dim], w_size);
724+ auto splited_dim_vec = split_parts (static_cast < int >( dims[dim]) , w_size);
725725 auto element_size = prec.size ();
726726
727727 VectorDims new_dims = dims;
@@ -740,7 +740,7 @@ MemoryPtr split_vertical(const dnnl::engine& eng,
740740 // total bytes
741741 auto mem_size = src->getSize ();
742742 // the steps need to copy.
743- const int step = (mem_size / channel_size);
743+ const int step = static_cast < int > (mem_size / channel_size);
744744 // bytes of selected dim.
745745 auto strideSize = splited_dim_vec[0 ] * element_size;
746746 auto copySize = splited_dim_vec[w_rank] * element_size;
@@ -749,8 +749,8 @@ MemoryPtr split_vertical(const dnnl::engine& eng,
749749 copySize /= 2 ;
750750 }
751751 cpu_parallel->parallel_for (step, [&](int i) {
752- int dst_offset = i * copySize;
753- int src_offset = i * splited_size + w_rank * strideSize;
752+ int dst_offset = static_cast < int >( i * copySize) ;
753+ int src_offset = static_cast < int >( i * splited_size + w_rank * strideSize) ;
754754 cpu_parallel_memcpy (dstPtr + dst_offset, srcPtr + src_offset, copySize);
755755 });
756756 return ptr;
0 commit comments