* Add Auto TensorCore TensorCore Unit Test * Rebase to tvm master branch & Add auto tensor core * Code Refine * Add tensor core switch by pragma * Add pragma in tensor core example code * Get real tile size to replace hard coded 16 * support more than 2 dimensions (e.g. batchmatmul) for buffer bind scope * support batch matmul * Move cuda env check to tensor_core.cc * Coderefine for tensor_core.cc * Refine comments * Some refinements of code and comment * Update TensorCore UT to pass the CPU test * remove redundant code * matmul's storage align for different layout * Add support for differenct position of type cast * Add formal tutorial for auto tensorcore codegen * move tensorcore check up to tutorial code * code and doc refine * comment out tune_and_evaluate in tutorial * fix cpplint error
Name |
Last commit
|
Last update |
---|---|---|
.. | ||
api | Loading commit data... | |
arithmetic | Loading commit data... | |
autotvm | Loading commit data... | |
codegen | Loading commit data... | |
common | Loading commit data... | |
contrib/hybrid | Loading commit data... | |
lang | Loading commit data... | |
node | Loading commit data... | |
op | Loading commit data... | |
pass | Loading commit data... | |
relay | Loading commit data... | |
runtime | Loading commit data... | |
schedule | Loading commit data... | |
README.md | Loading commit data... |