Minmin Sun (孙敏敏) d64bf6b51b Auto TensorCore CodeGen (#4234) ...

* 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

2019-11-09 13:01:36 -08:00

..

_ffi

[DOCS] Update link loc (#4257)

2019-11-05 16:03:04 -08:00

autotvm

[AutoTVM] Add batch_matmul to tunable operations (#4242)

2019-11-06 16:07:09 -08:00

contrib

[Contrib] Fix error message at callback_get_section_size() (#4221)

2019-11-06 10:39:40 -08:00

exec

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

hybrid

[ARITH] Use explicit div mode in python. (#4014)

2019-09-27 07:45:25 -07:00

micro

[WIP][µTVM] Add OpenOCD Low-Level Device (RISC-V Support) (#3756)

2019-09-02 15:32:52 +08:00

relay

[Relay][Frontend][ONNX] Add support for broadcasting to Where and MatMul (#4267)

2019-11-07 14:10:30 -08:00

rpc

[rpc] use callback func to do send & recv (#4147)

2019-10-23 11:02:34 -07:00

__init__.py

Quit and clean when TVM is interrupted (#3640)

2019-08-05 09:31:55 -07:00

_api_internal.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

_pyversion.py

[Relay] C++ GraphRuntimeCodegen, Deprecate Python2 (#2986)

2019-04-12 16:13:45 -07:00

api.py

[RUNTIME] Refactor object python FFI to new protocol. (#4128)

2019-10-16 15:24:23 -07:00

arith.py

[ARITH] Revamp IntSet (#3272)

2019-06-13 13:09:58 -07:00

attrs.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

build_module.py

Auto TensorCore CodeGen (#4234)

2019-11-09 13:01:36 -08:00

codegen.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

container.py

[DOC] various assorted grammar fixes (#3127)

2019-05-01 12:05:18 -04:00

datatype.py

[Datatypes] Custom datatypes (#2900)

2019-05-15 13:34:30 -07:00

error.py

[REFACTOR][NODE][RUNTIME] Move Node to the new Object protocol. (#4161)

2019-10-20 18:30:41 -07:00

expr.py

[ARITH] migrate indexdiv/mod to floordiv/mod (#4008)

2019-09-29 22:06:58 -07:00

generic.py

[ARITH] Use explicit div mode in python. (#4014)

2019-09-27 07:45:25 -07:00

intrin.py

Expose llvm.nearbyint intrinsic. This is a faster alternate to rounding. (#4001)

2019-09-25 11:22:54 -07:00

ir_builder.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

ir_pass.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

make.py

[ARITH] More recursive rewrite rule, cleanup simplify tests (#3502)

2019-07-06 22:48:43 -07:00

module.py

[TVM][AutoTVM] cast filepath arguments to string (#3968)

2019-09-18 08:12:32 -07:00

ndarray.py

Implementation of uTVM (#3227)

2019-07-25 10:12:57 -07:00

node.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

schedule.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

stmt.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

tag.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00

target.py

[TOPI] Added support for Mali Bifrost target (#4047)

2019-10-22 13:49:17 -07:00

tensor_intrin.py

Extend TensorComputeOp to allow scalar inputs (#2606). (#3300)

2019-06-21 21:22:54 -07:00

tensor.py

Add printer for Layout/BijectiveLayout (#3582)

2019-07-19 09:21:47 -07:00

testing.py

[HEADER] Add Header to Comply with ASF Release Policy (#2982)

2019-04-07 21:14:02 -07:00