FreeTensor
Loading...
Searching...
No Matches
Namespaces | Functions
lower_cutlass_micro_block.cc File Reference
#include <analyze/symbol_table.h>
#include <hash.h>
#include <mutator.h>
#include <numeric>
#include <pass/shrink_for.h>
#include <schedule/lower_cutlass_micro_block.h>
#include <schedule/var_merge.h>
#include <schedule/var_reorder.h>
#include <schedule/var_split.h>
#include <schedule/var_unsqueeze.h>
#include <type/data_type.h>
Include dependency graph for lower_cutlass_micro_block.cc:

Namespaces

namespace  freetensor
 

Functions

Stmt freetensor::lowerCutlassMicroBlock (const Stmt &ast, const ID &matMulId, const ID &defIdC, const std::vector< bool > &dimsCBatch, const std::vector< bool > &dimsCM, const std::vector< bool > &dimsCN)