OpSyncDevice.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 OpSyncDevice : public OpBase
{
  public:
    OpSyncDevice(const std::vector<std::shared_ptr<Memory>>& memObjects);
 
    OpSyncDevice(const OpSyncDevice&) = delete;
    OpSyncDevice(const OpSyncDevice&&) = delete;
    OpSyncDevice& operator=(const OpSyncDevice&) = delete;
    OpSyncDevice& operator=(const OpSyncDevice&&) = delete;
 
 
    ~OpSyncDevice() 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