aboutsummaryrefslogtreecommitdiff
path: root/compute_kernel_writer/validation/tests/CLTileTest.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'compute_kernel_writer/validation/tests/CLTileTest.hpp')
-rw-r--r--compute_kernel_writer/validation/tests/CLTileTest.hpp311
1 files changed, 311 insertions, 0 deletions
diff --git a/compute_kernel_writer/validation/tests/CLTileTest.hpp b/compute_kernel_writer/validation/tests/CLTileTest.hpp
new file mode 100644
index 0000000000..9fb47941f4
--- /dev/null
+++ b/compute_kernel_writer/validation/tests/CLTileTest.hpp
@@ -0,0 +1,311 @@
+#ifndef COMPUTE_KERNEL_WRITER_TESTS_CLTENSOR_HPP
+#define COMPUTE_KERNEL_WRITER_TESTS_CLTENSOR_HPP
+
+#include "src/Helpers.h"
+#include "src/cl/CLTile.h"
+#include "common/Common.h"
+
+#include <string>
+#include <vector>
+
+namespace ckw
+{
+class CLTileInternalVariableNamesTest : public ITest
+{
+public:
+ const int32_t width = 4;
+ const int32_t height = 4;
+ const DataType dt = DataType::Fp32;
+
+ CLTileInternalVariableNamesTest()
+ {
+ _tile_name.push_back("dst");
+ _tile_name.push_back("_G0_dst");
+ _tile_name.push_back("_SRC");
+ }
+
+ bool run() override
+ {
+ // The status of this variable can change in VALIDATE_TEST()
+ bool all_tests_passed = true;
+
+ const TileInfo info(dt, width, height);
+
+ const size_t num_tests = _tile_name.size();
+ for(size_t i = 0; i < num_tests; ++i)
+ {
+ const std::string tile_name = _tile_name[i];
+ const CLTile tile(tile_name, info);
+ const auto vars = tile.all();
+
+ for(int32_t y = 0; y < height; ++y)
+ {
+ const std::string expected_var_name = tile_name + "_" + std::to_string(y);
+ const std::string actual_var_name = vars[y].str;
+ VALIDATE_TEST(actual_var_name.compare(expected_var_name) == 0, all_tests_passed, i);
+ }
+ }
+ return all_tests_passed;
+ }
+
+ std::string name() override
+ {
+ return "CLTileInternalVariableNamesTest";
+ }
+
+private:
+ std::vector<std::string> _tile_name {};
+};
+
+class CLTileInternalNumVariablesTest : public ITest
+{
+public:
+ CLTileInternalNumVariablesTest()
+ {
+ _width.push_back(4);
+ _width.push_back(1);
+ _width.push_back(16);
+
+ _height.push_back(1);
+ _height.push_back(5);
+ _height.push_back(3);
+ }
+
+ bool run() override
+ {
+ VALIDATE_ON_MSG(_width.size() == _height.size(), "The number of widths and heights does not match");
+
+ // The status of this variable can change in VALIDATE_TEST()
+ bool all_tests_passed = true;
+
+ const size_t num_tests = _width.size();
+
+ for(size_t i = 0; i < num_tests; ++i)
+ {
+ const int32_t width = _width[i];
+ const int32_t height = _height[i];
+ const TileInfo info(DataType::Fp32, width, height);
+ const CLTile tile("src", info);
+ const auto vars = tile.all();
+ const int32_t num_vars = vars.size();
+
+ // We expect the number of variables to match the heigth of the tile
+ VALIDATE_TEST(num_vars == height, all_tests_passed, i);
+ }
+ return all_tests_passed;
+ }
+
+ std::string name() override
+ {
+ return "CLTileInternalNumVariablesTest";
+ }
+
+private:
+ std::vector<int32_t> _width {};
+ std::vector<int32_t> _height {};
+};
+
+class CLTileAccessScalarVariableTest : public ITest
+{
+public:
+ const std::string tile_name = "src";
+ const int32_t width = 16;
+ const int32_t height = 8;
+ const DataType dt = DataType::Fp32;
+
+ CLTileAccessScalarVariableTest()
+ {
+ _x_coord.push_back(4);
+ _x_coord.push_back(1);
+ _x_coord.push_back(15);
+ _x_coord.push_back(10);
+
+ _y_coord.push_back(1);
+ _y_coord.push_back(5);
+ _y_coord.push_back(3);
+ _y_coord.push_back(4);
+ }
+
+ bool run() override
+ {
+ const TileInfo info(dt, width, height);
+ const CLTile tile(tile_name, info);
+
+ VALIDATE_ON_MSG(_x_coord.size() == _y_coord.size(), "The number of x-coords and y-coords does not match");
+
+ // The status of this variable can change in VALIDATE_TEST()
+ bool all_tests_passed = true;
+
+ const size_t num_tests = _x_coord.size();
+
+ for(size_t i = 0; i < num_tests; ++i)
+ {
+ const int32_t x_coord = _x_coord[i];
+ const int32_t y_coord = _y_coord[i];
+
+ const TileVariable var = tile.scalar(x_coord, y_coord);
+
+ const std::string expected_var_name = var.str;
+ std::string actual_var_name = tile_name;
+ actual_var_name += "_" + std::to_string(y_coord);
+ actual_var_name += ".s" + dec_to_hex_as_string(x_coord);
+
+ VALIDATE_TEST(actual_var_name.compare(expected_var_name) == 0, all_tests_passed, i);
+ }
+ return all_tests_passed;
+ }
+
+ std::string name() override
+ {
+ return "CLTileAccessScalarVariableTest";
+ }
+
+private:
+ std::vector<int32_t> _x_coord {};
+ std::vector<int32_t> _y_coord {};
+};
+
+class CLTileAccessScalarVariableBroadcastXTest : public ITest
+{
+public:
+ const std::string tile_name = "src";
+ const int32_t height = 8;
+ const DataType dt = DataType::Fp32;
+
+ CLTileAccessScalarVariableBroadcastXTest()
+ {
+ _width.push_back(1);
+ _width.push_back(2);
+ _width.push_back(3);
+
+ _x_coord.push_back(4);
+ _x_coord.push_back(5);
+ _x_coord.push_back(6);
+
+ _y_coord.push_back(1);
+ _y_coord.push_back(3);
+ _y_coord.push_back(2);
+ }
+
+ bool run() override
+ {
+ VALIDATE_ON_MSG(_width.size() == _y_coord.size(), "The number of widths and y-coords does not match");
+ VALIDATE_ON_MSG(_x_coord.size() == _y_coord.size(), "The number of x-coords and y-coords does not match");
+
+ // The status of this variable can change in VALIDATE_TEST()
+ bool all_tests_passed = true;
+
+ const size_t num_tests = _x_coord.size();
+
+ for(size_t i = 0; i < num_tests; ++i)
+ {
+ const int32_t width = _width[i];
+ const int32_t x_coord = _x_coord[i];
+ const int32_t y_coord = _y_coord[i];
+
+ const int32_t x_coord_clamped = clamp(x_coord, static_cast<int32_t>(0), width - 1);
+
+ const TileInfo info(dt, width, height);
+ const CLTile tile(tile_name, info);
+
+ const TileVariable var = tile.scalar(x_coord, y_coord);
+
+ const std::string expected_var_name = var.str;
+ std::string actual_var_name = tile_name;
+ actual_var_name += "_" + std::to_string(y_coord);
+ if(width != 1)
+ {
+ actual_var_name += ".s" + dec_to_hex_as_string(x_coord_clamped);
+ }
+
+ VALIDATE_TEST(actual_var_name.compare(expected_var_name) == 0, all_tests_passed, i);
+ }
+ return all_tests_passed;
+ }
+
+ std::string name() override
+ {
+ return "CLTileAccessScalarVariableBroadcastXTest";
+ }
+
+private:
+ std::vector<int32_t> _width {};
+ std::vector<int32_t> _x_coord {};
+ std::vector<int32_t> _y_coord {};
+};
+
+class CLTileAccessScalarVariableBroadcastYTest : public ITest
+{
+public:
+ const std::string tile_name = "src";
+ const int32_t width = 8;
+ const DataType dt = DataType::Fp32;
+
+ CLTileAccessScalarVariableBroadcastYTest()
+ {
+ _height.push_back(1);
+ _height.push_back(2);
+ _height.push_back(3);
+
+ _x_coord.push_back(4);
+ _x_coord.push_back(5);
+ _x_coord.push_back(6);
+
+ _y_coord.push_back(3);
+ _y_coord.push_back(4);
+ _y_coord.push_back(5);
+ }
+
+ bool run() override
+ {
+ VALIDATE_ON_MSG(_height.size() == _y_coord.size(), "The number of widths and y-coords does not match");
+ VALIDATE_ON_MSG(_x_coord.size() == _y_coord.size(), "The number of x-coords and y-coords does not match");
+
+ // The status of this variable can change in VALIDATE_TEST()
+ bool all_tests_passed = true;
+
+ const size_t num_tests = _x_coord.size();
+
+ for(size_t i = 0; i < num_tests; ++i)
+ {
+ const int32_t height = _height[i];
+ const int32_t x_coord = _x_coord[i];
+ const int32_t y_coord = _y_coord[i];
+
+ const int32_t y_coord_clamped = clamp(y_coord, static_cast<int32_t>(0), height - 1);
+
+ const TileInfo info(dt, width, height);
+ const CLTile tile(tile_name, info);
+
+ const TileVariable var = tile.scalar(x_coord, y_coord);
+
+ const std::string expected_var_name = var.str;
+ std::string actual_var_name = tile_name;
+ if(height != 1)
+ {
+ actual_var_name += "_" + std::to_string(y_coord_clamped);
+ }
+
+ if(width != 1)
+ {
+ actual_var_name += ".s" + dec_to_hex_as_string(x_coord);
+ }
+
+ VALIDATE_TEST(actual_var_name.compare(expected_var_name) == 0, all_tests_passed, i);
+ }
+ return all_tests_passed;
+ }
+
+ std::string name() override
+ {
+ return "CLTileAccessScalarVariableBroadcastYTest";
+ }
+
+private:
+ std::vector<int32_t> _height {};
+ std::vector<int32_t> _x_coord {};
+ std::vector<int32_t> _y_coord {};
+};
+}
+
+#endif /* COMPUTE_KERNEL_WRITER_TESTS_CLTENSOR_HPP */
generated by cgit v1.2.1 (git 2.23.0) at 2024-05-28 19:21:44 +0000