fix uint32_t

Author: azhai219

src/plugins/intel_cpu/src/nodes/kernels/x64/gather_uni_kernel.hpp

@@ -87,9 +87,9 @@ struct jitGatherKernelBase {
     }
     explicit jitGatherKernelBase(const jGatherConfParams& jcp, uint64_t vlen, uint64_t indicesTypeSize)
         : jcp(jcp),
-          vlen(static_cast<uint32_t>(vlen)),
-          dataElPerVec(static_cast<uint32_t>(vlen / jcp.dataTypeSize)),
-          idxElPerVec(static_cast<uint32_t>(vlen / indicesTypeSize)),
+          vlen(vlen),
+          dataElPerVec(vlen / jcp.dataTypeSize),
+          idxElPerVec(vlen / indicesTypeSize),
           is_real16_to_f32((jcp.in_prec == element::f16 || jcp.in_prec == element::bf16) &&
                            jcp.out_prec == element::f32),
           is_f32_to_bf16(jcp.in_prec == element::f32 && jcp.out_prec == element::bf16) {}
@@ -109,9 +109,9 @@ struct jitGatherKernelBase {
 
 protected:
     jGatherConfParams jcp;
-    uint32_t vlen = 0;
-    uint32_t dataElPerVec = 0;
-    uint32_t idxElPerVec = 0;
+    uint64_t vlen = 0;
+    uint64_t dataElPerVec = 0;
+    uint64_t idxElPerVec = 0;
     static const unsigned shufMask8bitUni[16];
     static const unsigned permMask8bitA2[8];
     static const unsigned permMask8bitA5[16];

src/plugins/intel_cpu/src/nodes/kernels/x64/grid_sample.cpp

@@ -28,7 +28,7 @@ namespace ov::intel_cpu::kernel {
 
 template <x64::cpu_isa_t isa>
 GridSampleKernel<isa>::GridSampleKernel(const GridSampleKernelConfParams& jcp)
-    : GridSampleKernelBase(jit_name(), jcp, isa, static_cast<uint32_t>(x64::cpu_isa_traits_t<isa>::vlen)) {
+    : GridSampleKernelBase(jit_name(), jcp, isa, static_cast<uint64_t>(x64::cpu_isa_traits_t<isa>::vlen)) {
     if (dataTypeSize == 2) {
         dataTypeShift = 1;
     } else if (dataTypeSize == 4) {

src/plugins/intel_cpu/src/nodes/kernels/x64/grid_sample.hpp

@@ -91,23 +91,23 @@ class GridSampleKernelBase : public JitKernelBase {
           gridElPerVec(vlen / gridTypeSize) {}
 
     virtual void create_ker() = 0;
-    uint32_t getVecLen() const {
+    uint64_t getVecLen() const {
         return vlen;
     }
-    uint32_t getDataElPerVec() const {
+    uint64_t getDataElPerVec() const {
         return dataElPerVec;
     }
-    uint32_t getGridElPerVec() const {
+    uint64_t getGridElPerVec() const {
         return gridElPerVec;
     }
 
 protected:
     GridSampleKernelConfParams jcp;
-    uint32_t vlen = 16LU;
-    uint32_t dataTypeSize = 1LU;
-    uint32_t gridTypeSize = 1LU;
-    uint32_t dataElPerVec = 1LU;
-    uint32_t gridElPerVec = 1LU;
+    uint64_t vlen = 16LU;
+    uint64_t dataTypeSize = 1LU;
+    uint64_t gridTypeSize = 1LU;
+    uint64_t dataElPerVec = 1LU;
+    uint64_t gridElPerVec = 1LU;
 };
 
 template <dnnl::impl::cpu::x64::cpu_isa_t isa>

src/plugins/intel_cpu/src/nodes/kernels/x64/jit_kernel_base.cpp

@@ -419,10 +419,10 @@ void JitKernelBase::fillRestWorkMask(const Xbyak::Xmm& xmmDstMask,
     Xbyak::Label lEnd;
     auto r32Ones = getReg32();
     Xbyak::Reg64 r64Ones(r32Ones.getIdx());
-    auto elPerVec = static_cast<uint32_t>(x64::cpu_isa_traits_t<x64::sse41>::vlen / typeSize);
+    auto elPerVec = x64::cpu_isa_traits_t<x64::sse41>::vlen / typeSize;
 
     mov(r64Ones, 0xFFFFFFFFFFFFFFFF);
-    for (uint32_t i = 0; i < elPerVec; i++) {
+    for (uint64_t i = 0; i < elPerVec; i++) {
         cmp(rWorkRest, i);
         jle(lEnd, T_NEAR);
 
@@ -444,7 +444,7 @@ void JitKernelBase::fillRestWorkMask(const Xbyak::Ymm& ymmDstMask,
                                      const uint64_t typeSize) {
     OPENVINO_ASSERT(any_of(typeSize, 1U, 2U, 4U, 8U), "Could not fill data with type size ", typeSize);
     Xbyak::Label lEnd;
-    auto elPerVec = static_cast<uint32_t>(x64::cpu_isa_traits_t<x64::sse41>::vlen / typeSize);
+    auto elPerVec = x64::cpu_isa_traits_t<x64::sse41>::vlen / typeSize;
     auto r32Ones = getReg32();
     Xbyak::Reg64 r64Ones(r32Ones.getIdx());
     Xbyak::Xmm xmmDstMask(ymmDstMask.getIdx());
@@ -453,7 +453,7 @@ void JitKernelBase::fillRestWorkMask(const Xbyak::Ymm& ymmDstMask,
     uni_vpxor(ymmDstMask, ymmDstMask, ymmDstMask);
     for (uint8_t i = 0; i < 2; i++) {
         Xbyak::Label lPerm;
-        for (uint32_t j = 0; j < elPerVec; j++) {
+        for (uint64_t j = 0; j < elPerVec; j++) {
             cmp(rWorkRest, i * elPerVec + j);
             jle(i == 0 ? lEnd : lPerm, T_NEAR);
 

src/plugins/intel_cpu/src/nodes/strided_slice.cpp

@@ -318,7 +318,7 @@ void StridedSlice::initSupportedPrimitiveDescriptors() {
     }
     supportedTypes.push_back(LayoutType::ncsp);
     auto creators = BlockedDescCreator::getCommonCreators();
-    auto range = BlockedDescCreator::makeFilteredRange(creators, static_cast<uint32_t>(nDims), supportedTypes);
+    auto range = BlockedDescCreator::makeFilteredRange(creators, static_cast<unsigned int>(nDims), supportedTypes);
 
     for (auto itr = range.first; itr != range.second; ++itr) {
         config.inConfs[attrs.DATA_ID].setMemDesc(