Commit 549416f7 authored by LeiWang1999's avatar LeiWang1999
Browse files

Merge branch 'main' of https://github.com/microsoft/TileLang into main

parents 4d63633a 7fad4e88
This diff is collapsed.
This diff is collapsed.
......@@ -36,7 +36,8 @@ namespace tl {
using namespace tir;
For PartitionLoop(For op, Var thread_var, arith::Analyzer* analyzer, Fragment loop_layout);
For PartitionLoop(For op, Var thread_var, arith::Analyzer *analyzer,
Fragment loop_layout);
Fragment PlanLoopPartition(For op, size_t num_thread, int vectorize_size);
......
This diff is collapsed.
......@@ -35,11 +35,11 @@ namespace tl {
using namespace tir;
int GetVectorizeSize(const For& loop);
For VectorizeLoop(const For& loop, int vectorize_hint = -1);
int GetVectorizeSize(const For &loop);
For VectorizeLoop(const For &loop, int vectorize_hint = -1);
bool IndiceCanVectorize(PrimExpr expr, Var var, PrimExpr iter_var_size, int target_vectorized_size,
arith::Analyzer* analyzer);
bool IndiceCanVectorize(PrimExpr expr, Var var, PrimExpr iter_var_size,
int target_vectorized_size, arith::Analyzer *analyzer);
} // namespace tl
} // namespace tvm
......
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment