gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix inconsistent struct declaration instead of class. (#632)

Browse Source 
Looks like typo.
This commit is contained in:
Daniil Fukalov
2022-08-27 02:20:21 +03:00
committed by GitHub
parent 7394d732ad
commit fe0c29b9ec
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
include/triton/codegen/analysis/layout.h
View File

@@ -211,7 +211,8 @@ private:
TensorCoreType tensor_core_type_ = FP32_FP16_FP16_FP32; 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, scanline_layout(size_t num_warps,
const std::vector<int>& axes, const std::vector<int>& axes,
const std::vector<unsigned>& shape, const std::vector<unsigned>& shape,
@@ -225,7 +226,7 @@ struct scanline_layout: public distributed_layout {
int contig_per_thread(size_t k) { return nts_.at(k); } 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);} 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. // micro tile size. The size of a tile held by a thread block.
std::vector<int> mts_; std::vector<int> mts_;
// nano tile size. The size of a tile held by a thread. // nano tile size. The size of a tile held by a thread.