- 09 Nov, 2019 1 commit
-
-
* 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
Minmin Sun (孙敏敏) committed
-
- 24 Oct, 2019 1 commit
-
-
* add tensor core support * avoid memory bank conflict * fix thread sync & better performance * better performance * add schedule test for conv2d * extend into BatchMatMul * support config fragment shape and layout using intrinsic * add TensorCore tutorial * add int support and fix lint * address comment * add 32*16*8 TensorCore test * fix wmma include logic
Siyuan Feng committed
-
- 28 Sep, 2019 1 commit
-
-
Tianqi Chen committed
-
- 08 Apr, 2019 1 commit
-
-
* [HEADER] ASF header dir=include * [HEADER] ASF Header dir=src * [HEADER] ASF Header -dir=python * [HEADER] ASF header dir=topi * [HEADER] ASF Header dir=nnvm * [HEADER] ASF Header -dir=tutorials * [HEADER] ASF Header dir=tests * [HEADER] ASF Header -dir=docker * fix whitespace * [HEADER] ASF Header -dir=jvm * [HEADER] ASF Header -dir=web * [HEADER] ASF Header --dir=apps * [HEADER] ASF Header --dir=vta * [HEADER] ASF Header -dir=go * temp * [HEADER] ASF Header --dir=rust * [HEADER] Add ASF Header --dir=cmake * [HEADER] ASF Header --dir=docs * [HEADER] Header for Jenkinsfile * [HEADER] ASF Header to toml and md * [HEADER] ASF Header to gradle * Finalize rat cleanup * Fix permission * Fix java test * temporary remove nnvm onnx test
Tianqi Chen committed
-
- 11 Jan, 2019 1 commit
-
-
Tianqi Chen committed
-
- 29 Dec, 2018 1 commit
-
-
Tianqi Chen committed
-
- 21 Oct, 2018 1 commit
-
-
Sergey Mironov committed
-
- 11 Sep, 2018 1 commit
-
-
Jian Weng committed
-
- 15 Jul, 2018 1 commit
-
-
Lianmin Zheng committed
-
- 29 May, 2018 1 commit
-
-
tqchen committed
-
- 17 Mar, 2018 1 commit
-
-
Pariksheet Pinjari committed
-
- 23 Jan, 2018 2 commits
-
-
Tianqi Chen committed
-
yuruofeifei committed
-
- 09 Jan, 2018 1 commit
-
-
* small fixs on docs * add IR output after parallelization
Yida Wang committed
-
- 08 Jan, 2018 1 commit
-
-
* Improve opt_gemm tutorial * Addressed comments
yuruofeifei committed
-
- 25 Sep, 2017 1 commit
-
-
Salem Derisavi committed
-
- 29 Aug, 2017 1 commit
-
-
Tianqi Chen committed
-