github-featured/apache--tvm
2
0
Fork 0
Code Issues Pull Requests Deployments Models Agent Notes Packages Projects Releases Wiki Activity
8,558 Commits 30 Branches 63 Tags
d0da0b94dea206400d3bf4e15cb7815713c5b6e7
Commit Graph

4 Commits

Author SHA1 Message Date
Jared Roesch f13fed55cf [Format] Convert all Python code w/o CI (#6448) 
* Add black setup

* Tweak pyproject.toml

* Fix syntax issues

* Fix

* Tweak

* Black all Python code
 2020-09-11 22:17:24 +09:00
Tianqi Chen f143881314 [PYTHON] Enhance with_attr API, cleanup MakeAPILegacy in testcases (#5335) 2020-04-15 11:11:28 -07:00
Tianqi Chen 75e936e1b5 [REFACTOR][TIR] Migrate most of low-level build to use the Pass Manager. (#5225) 
* [REFACTOR][TIR] Migrate most of low-level build to use the Pass Manager.

- SplitHostDevice
- ThreadSync
- BindDevice
- LowerThreadAllreduce
- Provide a temp fix for printing IRModule with PrimFunc before the formal text printer.

* Address comments, fix tests.

* Fix relay tests

* Explicit move
 2020-04-03 15:50:11 -07:00
Tianqi Chen e03164159c [TIR] Introduce tir::PrimFunc (#5070) 
This PR introduces tir::PrimFunc which will be used as the TIR function
container in the unified IR.

Also streamlined the function attributes a bit further.
- All common attributes are under tvm::attr
- TIR specific attributes are under tvm::tir::attr and comes with a tir prefix
- Use stl_style for attributes for now
 2020-03-14 14:26:23 -07:00