diff --git a/single_include/kompute/Kompute.hpp b/single_include/kompute/Kompute.hpp index d1360f5..1248a35 100755 --- a/single_include/kompute/Kompute.hpp +++ b/single_include/kompute/Kompute.hpp @@ -1647,7 +1647,7 @@ public: */ Algorithm(std::shared_ptr device, std::shared_ptr commandBuffer, - const std::vector& specializationConstants = {}); + const Constants& specializationConstants = {}); /** * Initialiser for the shader data provided to the algorithm as well as @@ -1699,7 +1699,7 @@ private: bool mFreePipeline = false; // -------------- ALWAYS OWNED RESOURCES - std::vector mSpecializationConstants; + Constants mSpecializationConstants; // Create util functions void createShaderModule(const std::vector& shaderFileData); @@ -1746,7 +1746,7 @@ class OpAlgoBase : public OpBase std::shared_ptr commandBuffer, std::vector>& tensors, const Workgroup& komputeWorkgroup = {}, - const std::vector& specializationConstants = {}); + const Constants& specializationConstants = {}); /** * Constructor that enables a file to be passed to the operation with @@ -1766,7 +1766,7 @@ class OpAlgoBase : public OpBase std::vector>& tensors, std::string shaderFilePath, const Workgroup& komputeWorkgroup = {}, - const std::vector& specializationConstants = {}); + const Constants& specializationConstants = {}); /** * Constructor that enables raw shader data to be passed to the main operation @@ -1785,7 +1785,7 @@ class OpAlgoBase : public OpBase std::vector>& tensors, const std::vector& shaderDataRaw, const Workgroup& komputeWorkgroup = {}, - const std::vector& specializationConstants = {}); + const Constants& specializationConstants = {}); /** * Default destructor, which is in charge of destroying the algorithm