streamk_common.hpp File Reference#
streamk_common.hpp File Reference
#include "ck_tile/core.hpp"Go to the source code of this file.
Namespaces | |
| namespace | ck_tile |
Enumerations | |
| enum | ck_tile::StreamKReductionStrategy : uint32_t { ck_tile::Atomic = 0u , ck_tile::Reduction = 1u } |
Functions | |
| template<ck_tile::StreamKReductionStrategy ReductionStrategy> | |
| ck_tile::index_t | ck_tile::estimate_num_wgs_per_tile (index_t sk_ctas, index_t iters_per_sk_cta, index_t iters_per_tile) |
| Estimates the number of Stream-K workgroups per macro tile in the C tensor. | |