Auto TensorCore CodeGen - #4234

Merged

tqchen merged 21 commits into

apache:masterfrom

minminsun:AutoTensorCoreCodeGen

Nov 9, 2019

Commits on Oct 31, 2019

Add Auto TensorCore TensorCore Unit Test
minminsun
committed
bfffc34
Rebase to tvm master branch & Add auto tensor core

jcf94
authored and
minminsun
committed
8772a03
Code Refine

jcf94
authored and
minminsun
committed
ebb01b7
Add tensor core switch by pragma

jcf94
authored and
minminsun
committed
626166b
Add pragma in tensor core example code

jcf94
authored and
minminsun
committed
2f7ab5b
Get real tile size to replace hard coded 16
minminsun
committed
e0cbc0e
support more than 2 dimensions (e.g. batchmatmul) for buffer bind scope
minminsun
committed
af0824b
support batch matmul

Orion34-lanbo
authored and
minminsun
committed
a76e4e7
Move cuda env check to tensor_core.cc

jcf94
authored and
minminsun
committed
4b0d18d
Coderefine for tensor_core.cc

jcf94
authored and
minminsun
committed
f80fad8
Refine comments
minminsun
committed
afa9061

Commits on Nov 6, 2019

Some refinements of code and comment

jcf94
authored and
minminsun
committed
7c6b880
Update TensorCore UT to pass the CPU test
jcf94
committed
d5134f2

Commits on Nov 7, 2019

Commits on Nov 8, 2019

code and doc refine
minminsun
committed
9baf969
comment out tune_and_evaluate in tutorial
minminsun
committed
f61f362
fix cpplint error
minminsun
committed
4824579