Skip to content

Commit

Permalink
Tensor Expression Debug Display (TEDD) (apache#4651)
Browse files Browse the repository at this point in the history
* Initial TEDD for publishing.

* 1. Fix lint issues. 2. Print intrin.body instead of intrin.name in Schedule Tree.  3. Add examples to top level APIs' comments.  4. Top level APIs don't print Dot string by default, unless outputdotstring is True.

* Fix more lint issues.

* Update top level API argument names and use raw strings to avoid Python lint warnings in the tests.

* Disable TEDD verification, but keep TE construction.

* Stop importing tedd to avoid failure.

* Separate data extraction and visualization. 1. Add API tedd.dump_json(schedule) to dump a json string for the schedule data for visualization.  2. Update tests.  3. Add a tutorial.  4. Add range information to IterVars.

* Update TEDD about InferBound failure.  1. TEDD doesn't call inferbound for DFG. 2. Update tutorial about the InferBound failure.

* 1. Import IPython only if SVG is requested.  This is required to fix a tutorial publishing faliure.  2. Fix test about IPython availability check.
  • Loading branch information
yongfeng-nv authored and alexwong committed Feb 28, 2020
1 parent 59f0bdf commit 146a7b8
Show file tree
Hide file tree
Showing 3 changed files with 1,039 additions and 0 deletions.
Loading

0 comments on commit 146a7b8

Please sign in to comment.