58 lines
1.6 KiB
C++
58 lines
1.6 KiB
C++
// 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 {
|
|
|
|
/**
|
|
* Operation that performs multiplication on two tensors and outpus on third
|
|
* tensor.
|
|
*/
|
|
class OpMult : public OpAlgoDispatch
|
|
{
|
|
public:
|
|
/**
|
|
* Default constructor with parameters that provides the bare minimum
|
|
* requirements for the operations to be able to create and manage their
|
|
* sub-components.
|
|
*
|
|
* @param tensors Tensors that are to be used in this operation
|
|
* @param algorithm An algorithm that will be overridden with the OpMult
|
|
* shader data and the tensors provided which are expected to be 3
|
|
*/
|
|
OpMult(std::vector<std::shared_ptr<Tensor>> tensors,
|
|
std::shared_ptr<Algorithm> algorithm)
|
|
: OpAlgoDispatch(algorithm)
|
|
{
|
|
KP_LOG_DEBUG("Kompute OpMult constructor with params");
|
|
|
|
if (tensors.size() != 3) {
|
|
throw std::runtime_error(
|
|
"Kompute OpMult expected 3 tensors but got " +
|
|
std::to_string(tensors.size()));
|
|
}
|
|
|
|
const std::vector<uint32_t> spirv = std::vector<uint32_t>(
|
|
SHADEROPMULT_COMP_SPV.begin(), SHADEROPMULT_COMP_SPV.end());
|
|
|
|
algorithm->rebuild<>(tensors, spirv);
|
|
}
|
|
|
|
/**
|
|
* Default destructor, which is in charge of destroying the algorithm
|
|
* components but does not destroy the underlying tensors
|
|
*/
|
|
~OpMult() override { KP_LOG_DEBUG("Kompute OpMult destructor started"); }
|
|
};
|
|
|
|
} // End namespace kp
|