aboutsummaryrefslogtreecommitdiff
path: root/compute_kernel_writer/prototype/include/ckw
diff options
context:
space:
mode:
authorNikolaj Jensen <nikolaj.jensen@arm.com>2023-07-03 09:44:42 +0100
committerNikolaj Jensen <nikolaj.jensen@arm.com>2023-07-07 08:51:11 +0000
commitacea4071a7f457bab696dc3c895ba47d60345541 (patch)
tree66a5f659f995ea110bcb8e217dc6a94e1f7ac47d /compute_kernel_writer/prototype/include/ckw
parente86f992d26a79cad76244c4444d113e45afa9b88 (diff)
downloadComputeLibrary-acea4071a7f457bab696dc3c895ba47d60345541.tar.gz
Fix code formatting in CKW
Signed-off-by: Nikolaj Jensen <nikolaj.jensen@arm.com> Change-Id: I8064b345c1efd243f8bded12ed5d561afe7c339a Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9854 Benchmark: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Jakub Sujak <jakub.sujak@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'compute_kernel_writer/prototype/include/ckw')
-rw-r--r--compute_kernel_writer/prototype/include/ckw/KernelWriter.h1
-rw-r--r--compute_kernel_writer/prototype/include/ckw/OperandBase.h1
-rw-r--r--compute_kernel_writer/prototype/include/ckw/TensorInfo.h10
-rw-r--r--compute_kernel_writer/prototype/include/ckw/TensorOperand.h2
-rw-r--r--compute_kernel_writer/prototype/include/ckw/TileInfo.h8
5 files changed, 20 insertions, 2 deletions
diff --git a/compute_kernel_writer/prototype/include/ckw/KernelWriter.h b/compute_kernel_writer/prototype/include/ckw/KernelWriter.h
index a2778a9485..3b1539116a 100644
--- a/compute_kernel_writer/prototype/include/ckw/KernelWriter.h
+++ b/compute_kernel_writer/prototype/include/ckw/KernelWriter.h
@@ -39,6 +39,7 @@ namespace ckw
namespace prototype
{
struct GpuKernelWriterAttribute;
+
class IGpuKernelWriter;
} // namespace prototype
diff --git a/compute_kernel_writer/prototype/include/ckw/OperandBase.h b/compute_kernel_writer/prototype/include/ckw/OperandBase.h
index f4825e16a7..a9e313fc0a 100644
--- a/compute_kernel_writer/prototype/include/ckw/OperandBase.h
+++ b/compute_kernel_writer/prototype/include/ckw/OperandBase.h
@@ -33,6 +33,7 @@ namespace ckw
namespace prototype
{
class IGpuKernelWriter;
+
class Operand;
} // namespace prototype
diff --git a/compute_kernel_writer/prototype/include/ckw/TensorInfo.h b/compute_kernel_writer/prototype/include/ckw/TensorInfo.h
index 00bb60a444..807158896b 100644
--- a/compute_kernel_writer/prototype/include/ckw/TensorInfo.h
+++ b/compute_kernel_writer/prototype/include/ckw/TensorInfo.h
@@ -117,20 +117,28 @@ public:
* - less than 0: bind a virtual tensor (tile)
*/
TensorInfo(DataType dt, const TensorShape &shape, TensorDataLayout dl, int32_t id);
+
/** Set shape */
TensorInfo &shape(const TensorShape &shape);
+
/** Get shape */
TensorShape shape() const;
+
/** Set data type */
TensorInfo &data_type(DataType dt);
+
/** Get data type */
DataType data_type() const;
+
/** Set data layout */
TensorInfo &data_layout(TensorDataLayout dl);
+
/** Get data layout */
TensorDataLayout data_layout() const;
+
/** Set id */
TensorInfo &id(int32_t id);
+
/** Get layout */
int32_t id() const;
@@ -140,6 +148,6 @@ private:
TensorDataLayout _dl{ TensorDataLayout::Unknown };
int32_t _id{ -1 };
};
-} // namespace kw
+} // namespace ckw
#endif /* CKW_PROTOTYPE_INCLUDE_CKW_TENSORINFO_H */
diff --git a/compute_kernel_writer/prototype/include/ckw/TensorOperand.h b/compute_kernel_writer/prototype/include/ckw/TensorOperand.h
index 2fc5044d1c..7a663f095b 100644
--- a/compute_kernel_writer/prototype/include/ckw/TensorOperand.h
+++ b/compute_kernel_writer/prototype/include/ckw/TensorOperand.h
@@ -134,7 +134,7 @@ public:
private:
TensorInfo _info;
- TileOperand *_tile{ nullptr };
+ TileOperand *_tile{ nullptr };
TensorTileSampler _tile_sampler{};
::std::unique_ptr<TensorComponentOperand> _stride1{ nullptr };
diff --git a/compute_kernel_writer/prototype/include/ckw/TileInfo.h b/compute_kernel_writer/prototype/include/ckw/TileInfo.h
index 8fba8bb827..c60880dcd1 100644
--- a/compute_kernel_writer/prototype/include/ckw/TileInfo.h
+++ b/compute_kernel_writer/prototype/include/ckw/TileInfo.h
@@ -48,12 +48,14 @@ public:
* @param[in] dt Tile data type
*/
TileInfo(DataType dt);
+
/** Constructor used to initialize a vector with a given data type and vector length.
*
* @param[in] dt Tile data type
* @param[in] w Tile width (or vector length)
*/
TileInfo(DataType dt, int32_t w);
+
/** Constructor used to initialize a tile with a given data type and tile sizes.
*
* @param[in] dt Tile data type
@@ -61,16 +63,22 @@ public:
* @param[in] w Tile width
*/
TileInfo(DataType dt, int32_t h, int32_t w);
+
/** Set width */
TileInfo &width(int32_t w);
+
/** Get width */
int32_t width() const;
+
/** Set height */
TileInfo &height(int32_t h);
+
/** Get height */
int32_t height() const;
+
/** Set data type */
TileInfo &data_type(DataType dt);
+
/** Get data type */
DataType data_type() const;