From fe0c29b9ece6e9f88f4d4ed9612c3d591416f821 Mon Sep 17 00:00:00 2001 From: Daniil Fukalov <1671137+dfukalov@users.noreply.github.com> Date: Sat, 27 Aug 2022 02:20:21 +0300 Subject: [PATCH] Fix inconsistent struct declaration instead of class. (#632) Looks like typo. --- include/triton/codegen/analysis/layout.h | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/include/triton/codegen/analysis/layout.h b/include/triton/codegen/analysis/layout.h index a69687875..313c7b1b3 100644 --- a/include/triton/codegen/analysis/layout.h +++ b/include/triton/codegen/analysis/layout.h @@ -211,7 +211,8 @@ private: TensorCoreType tensor_core_type_ = FP32_FP16_FP16_FP32; }; -struct scanline_layout: public distributed_layout { +class scanline_layout: public distributed_layout { +public: scanline_layout(size_t num_warps, const std::vector& axes, const std::vector& shape, @@ -225,7 +226,7 @@ struct scanline_layout: public distributed_layout { int contig_per_thread(size_t k) { return nts_.at(k); } int per_thread(size_t k) { return contig_per_thread(k) * shape_[k] / shape_per_cta(k);} -public: +private: // micro tile size. The size of a tile held by a thread block. std::vector mts_; // nano tile size. The size of a tile held by a thread.