diff --git a/single_include/kompute/Kompute.hpp b/single_include/kompute/Kompute.hpp
index dc85ab797..2f59d7fe5 100755
--- a/single_include/kompute/Kompute.hpp
+++ b/single_include/kompute/Kompute.hpp
@@ -1442,11 +1442,13 @@ OpAlgoLhsRhsOut<tX, tY, tZ>::record()
 
     // Barrier to ensure the data is finished writing to buffer memory
     this->mTensorLHS->recordBufferMemoryBarrier(
+      this->mCommandBuffer,
       vk::AccessFlagBits::eHostWrite,
       vk::AccessFlagBits::eShaderRead,
       vk::PipelineStageFlagBits::eHost,
       vk::PipelineStageFlagBits::eComputeShader);
     this->mTensorRHS->recordBufferMemoryBarrier(
+      this->mCommandBuffer,
       vk::AccessFlagBits::eHostWrite,
       vk::AccessFlagBits::eShaderRead,
       vk::PipelineStageFlagBits::eHost,
@@ -1456,6 +1458,7 @@ OpAlgoLhsRhsOut<tX, tY, tZ>::record()
 
     // Barrier to ensure the shader code is executed before buffer read
     this->mTensorOutput->recordBufferMemoryBarrier(
+      this->mCommandBuffer,
       vk::AccessFlagBits::eShaderWrite,
       vk::AccessFlagBits::eTransferRead,
       vk::PipelineStageFlagBits::eComputeShader,
diff --git a/src/include/kompute/operations/OpAlgoLhsRhsOut.hpp b/src/include/kompute/operations/OpAlgoLhsRhsOut.hpp
index e513bb820..41f8c103e 100644
--- a/src/include/kompute/operations/OpAlgoLhsRhsOut.hpp
+++ b/src/include/kompute/operations/OpAlgoLhsRhsOut.hpp
@@ -177,11 +177,13 @@ OpAlgoLhsRhsOut<tX, tY, tZ>::record()
 
     // Barrier to ensure the data is finished writing to buffer memory
     this->mTensorLHS->recordBufferMemoryBarrier(
+      this->mCommandBuffer,
       vk::AccessFlagBits::eHostWrite,
       vk::AccessFlagBits::eShaderRead,
       vk::PipelineStageFlagBits::eHost,
       vk::PipelineStageFlagBits::eComputeShader);
     this->mTensorRHS->recordBufferMemoryBarrier(
+      this->mCommandBuffer,
       vk::AccessFlagBits::eHostWrite,
       vk::AccessFlagBits::eShaderRead,
       vk::PipelineStageFlagBits::eHost,
@@ -191,6 +193,7 @@ OpAlgoLhsRhsOut<tX, tY, tZ>::record()
 
     // Barrier to ensure the shader code is executed before buffer read
     this->mTensorOutput->recordBufferMemoryBarrier(
+      this->mCommandBuffer,
       vk::AccessFlagBits::eShaderWrite,
       vk::AccessFlagBits::eTransferRead,
       vk::PipelineStageFlagBits::eComputeShader,