OpTensorSyncDevice.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 OpTensorSyncDevice : public OpBase
{
  public:
    OpTensorSyncDevice(const std::vector<std::shared_ptr<Tensor>>& tensors);
 
    ~OpTensorSyncDevice() 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<Tensor>> mTensors;
};
 
} // End namespace kp