OpMemoryBarrier.hpp

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include "kompute/Algorithm.hpp"
#include "kompute/Core.hpp"
#include "kompute/Tensor.hpp"
#include "kompute/operations/OpBase.hpp"
 
namespace kp {
 
class OpMemoryBarrier : public OpBase
{
  public:
    OpMemoryBarrier(const std::vector<std::shared_ptr<Memory>>& memObjects,
                    const vk::AccessFlagBits& srcAccessMask,
                    const vk::AccessFlagBits& dstAccessMask,
                    const vk::PipelineStageFlagBits& srcStageMask,
                    const vk::PipelineStageFlagBits& dstStageMask,
                    bool barrierOnPrimary = true) noexcept;
 
    OpMemoryBarrier(const OpMemoryBarrier&) = delete;
    OpMemoryBarrier(const OpMemoryBarrier&&) = delete;
    OpMemoryBarrier& operator=(const OpMemoryBarrier&) = delete;
    OpMemoryBarrier& operator=(const OpMemoryBarrier&&) = delete;
 
    virtual ~OpMemoryBarrier() noexcept override;
 
    virtual void record(const vk::CommandBuffer& commandBuffer) override;
 
    virtual void preEval(const vk::CommandBuffer& commandBuffer) override;
 
    virtual void postEval(const vk::CommandBuffer& commandBuffer) override;
 
  private:
    const vk::AccessFlagBits mSrcAccessMask;
    const vk::AccessFlagBits mDstAccessMask;
    const vk::PipelineStageFlagBits mSrcStageMask;
    const vk::PipelineStageFlagBits mDstStageMask;
    const bool mBarrierOnPrimary;
    const std::vector<std::shared_ptr<Memory>> mMemObjects;
};
 
} // End namespace kp