OpMult.hpp

// SPDX-License-Identifier: Apache-2.0
#pragma once
 
#include <fstream>
 
#include "kompute/Core.hpp"
 
#include "ShaderOpMult.hpp"
 
#include "kompute/Algorithm.hpp"
#include "kompute/Tensor.hpp"
 
#include "kompute/operations/OpAlgoDispatch.hpp"
 
namespace kp {
 
class OpMult : public OpAlgoDispatch
{
  public:
    OpMult(std::vector<std::shared_ptr<Memory>> memObjects,
           std::shared_ptr<Algorithm> algorithm)
      : OpAlgoDispatch(algorithm)
    {
        KP_LOG_DEBUG("Kompute OpMult constructor with params");
 
        if (memObjects.size() != 3) {
            throw std::runtime_error(
              "Kompute OpMult expected 3 mem objects but got " +
              std::to_string(memObjects.size()));
        }
 
        const std::vector<uint32_t> spirv = std::vector<uint32_t>(
          SHADEROPMULT_COMP_SPV.begin(), SHADEROPMULT_COMP_SPV.end());
 
        algorithm->rebuild<>(memObjects, spirv);
    }
 
    OpMult(const OpMult&) = delete;
    OpMult(const OpMult&&) = delete;
    OpMult& operator=(const OpMult&) = delete;
    OpMult& operator=(const OpMult&&) = delete;
 
    ~OpMult() noexcept override { KP_LOG_DEBUG("Kompute OpMult destructor started"); }
};
 
} // End namespace kp