FreeTensor
Loading...
Searching...
No Matches
lower_cutlass_micro_block.h
Go to the documentation of this file.
1#ifndef FREE_TENSOR_LOWER_CUTLASS_MICRO_BLOCK_H
2#define FREE_TENSOR_LOWER_CUTLASS_MICRO_BLOCK_H
3
4#include <stmt.h>
5
6namespace freetensor {
7
8Stmt lowerCutlassMicroBlock(const Stmt &ast, const ID &matMulId,
9 const ID &defIdC,
10 const std::vector<bool> &dimsCBatch,
11 const std::vector<bool> &dimsCM,
12 const std::vector<bool> &dimsCN);
13
14}
15
16#endif // FREE_TENSOR_LOWER_CUTLASS_MICRO_BLOCK_H
Definition: allocator.h:9
Ref< StmtNode > Stmt
Definition: ast.h:152
Stmt 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)
Definition: lower_cutlass_micro_block.cc:328