OpSyncLocal.hpp

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include "kompute/Core.hpp"
 
#include "kompute/Tensor.hpp"
 
#include "kompute/operations/OpBase.hpp"
 
namespace kp {
 
class OpSyncLocal : public OpBase
{
  public:
    OpSyncLocal(const std::vector<std::shared_ptr<Memory>>& memObjects);
 
    OpSyncLocal(const OpSyncLocal&) = delete;
    OpSyncLocal(const OpSyncLocal&&) = delete;
    OpSyncLocal& operator=(const OpSyncLocal&) = delete;
    OpSyncLocal& operator=(const OpSyncLocal&&) = delete;
 
    ~OpSyncLocal() noexcept override;
 
    void record(const vk::CommandBuffer& commandBuffer) override;
 
    virtual void preEval(const vk::CommandBuffer& commandBuffer) override;
 
    virtual void postEval(const vk::CommandBuffer& commandBuffer) override;
 
  private:
    // -------------- ALWAYS OWNED RESOURCES
    std::vector<std::shared_ptr<Memory>> mMemObjects;
};
 
} // End namespace kp