FreeTensor
Loading...
Searching...
No Matches
include
schedule
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
6
namespace
freetensor
{
7
8
Stmt
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
freetensor
Definition:
allocator.h:9
freetensor::Stmt
Ref< StmtNode > Stmt
Definition:
ast.h:152
freetensor::lowerCutlassMicroBlock
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
stmt.h
Generated by
1.9.6