diff --git a/rocclr/device/blit.hpp b/rocclr/device/blit.hpp
index ec98886b38..cc4ddda2dc 100644
--- a/rocclr/device/blit.hpp
+++ b/rocclr/device/blit.hpp
@@ -236,7 +236,7 @@ class BlitManager : public amd::HeapObject {
   //! Stream batch memory operation
   virtual bool batchMemOps(const void* paramArray,
                            size_t paramSize,
-                           uint32_t count
+                           uint64_t count
   ) const = 0;
 
   //! Enables synchronization on blit operations
diff --git a/rocclr/device/pal/palblit.hpp b/rocclr/device/pal/palblit.hpp
index 7df8731328..0d73ba632c 100644
--- a/rocclr/device/pal/palblit.hpp
+++ b/rocclr/device/pal/palblit.hpp
@@ -478,7 +478,7 @@ class KernelBlitManager : public DmaBlitManager {
   //! Batch memory ops- Submits batch of streamWaits and streamWrite operations.
   virtual bool batchMemOps(const void* paramArray,
                            size_t paramSize,
-                           uint32_t count) const {
+                           uint64_t count) const {
     assert(!"Unimplemented");
     return false;
   }
diff --git a/rocclr/device/rocm/rocblit.cpp b/rocclr/device/rocm/rocblit.cpp
index a9ffb4d95a..5ef076673d 100644
--- a/rocclr/device/rocm/rocblit.cpp
+++ b/rocclr/device/rocm/rocblit.cpp
@@ -2513,7 +2513,7 @@ bool KernelBlitManager::streamOpsWait(device::Memory& memory, uint64_t value, si
 
 // ================================================================================================
 bool KernelBlitManager::batchMemOps(const void* paramArray, size_t paramSize,
-                                    uint32_t count) const {
+                                    uint64_t count) const {
   amd::ScopedLock k(lockXferOps_);
   bool result = false;
   uint blitType = BatchMemOp;
diff --git a/rocclr/device/rocm/rocblit.hpp b/rocclr/device/rocm/rocblit.hpp
index 1d9d0e670d..6a979ac5b4 100644
--- a/rocclr/device/rocm/rocblit.hpp
+++ b/rocclr/device/rocm/rocblit.hpp
@@ -505,7 +505,7 @@ class KernelBlitManager : public DmaBlitManager {
   ) const;
 
   //! Batch memory ops- Submits batch of streamWaits and streamWrite operations.
-  virtual bool batchMemOps(const void* paramArray, size_t paramSize, uint32_t count) const;
+  virtual bool batchMemOps(const void* paramArray, size_t paramSize, uint64_t count) const;
 
   virtual amd::Monitor* lockXfer() const { return &lockXferOps_; }