diff options
Diffstat (limited to 'compute_kernel_writer/src/KernelWriter.cpp')
| -rw-r--r-- | compute_kernel_writer/src/KernelWriter.cpp | 17 |
1 files changed, 10 insertions, 7 deletions
diff --git a/compute_kernel_writer/src/KernelWriter.cpp b/compute_kernel_writer/src/KernelWriter.cpp index 347ae5b545..7b83eade6f 100644 --- a/compute_kernel_writer/src/KernelWriter.cpp +++ b/compute_kernel_writer/src/KernelWriter.cpp @@ -22,14 +22,13 @@ * SOFTWARE. */ +#include "ckw/KernelWriter.h" #include "ckw/Error.h" #include "ckw/TileOperand.h" -#include "ckw/KernelWriter.h" #include "ckw/types/TargetArchitecture.h" #include "ckw/types/TargetLanguage.h" #include "src/cl/CLKernelWriter.h" -#include <iterator> namespace ckw { @@ -55,15 +54,19 @@ int32_t KernelWriter::id_space() const return _id_space; } -TileOperand &KernelWriter::add_operand(std::unique_ptr<TileOperand> &operand_ptr) +std::string KernelWriter::generate_full_name(const std::string &name) const { - auto it = _operands.insert(std::move(operand_ptr)); - return *it.first->get(); + return "G" + std::to_string(id_space()) + "__" + name; } -std::string KernelWriter::generate_full_name(const std::string &name) const +TileOperand KernelWriter::create_tile_operand(ITile &tile) { - return "G" + std::to_string(id_space()) + "__" + name; + return TileOperand(tile); +} + +ITile &KernelWriter::get_tile(const TileOperand &operand) +{ + return operand._tile; } } // namespace ckw |