Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Meta Schedule] Add Replay Tuning Interface #543

Merged
merged 15 commits into from
Dec 10, 2021
Merged

[Meta Schedule] Add Replay Tuning Interface #543

merged 15 commits into from
Dec 10, 2021

Commits on Dec 9, 2021

  1. Squashed commit 

    [Meta Schedule][M3c] Schedule Rules, Mutator & Postprocs (#485)

[Meta Schedule][M3c] PostOrderApply (#486)

Fix Post Order Apply (#490)

[MetaSchedule] Relay Integration (#489)

[M3c][Meta Schedule] Add Trace Correctness Test for PostOrderApply (#492)

Fix replay trace. (#493)

[M3c][Meta Schedule] Implement the Replay Func class. (#495)

[PR] Test script for meta-schedule task extraction. Interface to load… (#494)

[Meta Schedule Refactor] Get child blocks (#500)

Read-at && Write-at (#497)

[M3c][Meta Schedule] Measure Callbacks (#498)

[Bug] Fix Infinite Loop Caused When Calling Methods Not Overrided In PyClass (#496)

[MetaSchedule] Sample-Perfect-Tile (#501)

[MetaSchedule] TE Workloads (#502)

[TensorIR] GetProducer, GetConsumer (#506)

[MetaScheduleRefactor] Annotate&Unannotate (#505)

[MetaSchedule] Multi-Level-Tiling & Auto-Inline (#503)

[Tests] Add unittests for auto-inline and multi-level-tiling (#508)

[Meta Schedule] Minor Fixes (#507)

[MetaSchedule] Rewrite Cooperative-Fetching / Unbound-Block / Reduction-Block (#509)

[MetaSchedule] Rewrite Parallel-Vectorize-Unroll / Verify-GPU / Disallow-Dynamic-Loops (#499)

[Meta Schedule] Add Helper Function & Minor Modification (#512)

[MetaSchedule] Test for Rewrite Parallel-Vectorize-Unroll  (#513)

[Meta Schedule] Feature Extractor & Cost Model (#510)

Blockize & Tensorize (#514)

Layout Rewriting: Suggest-Index-Map (#520)

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Sunghyun Park <49998730+sunggg@users.noreply.github.com>
    @junrushao @Hzfengsy
    @spectrometerHBH @jinhongyii @MasterJH5574 @vinx13 @sunggg @zxybazh
    7 people authored and zxybazh committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    14e5793 View commit details
    Browse the repository at this point in the history

  2. [MetaSchedule] Parallel-Vectorize-Unroll & Random-Compute-Location (#… 

    …516)

* parallel vectorize unroll & random compute location

* rebased
    @spectrometerHBH @zxybazh
    spectrometerHBH authored and zxybazh committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    7877862 View commit details
    Browse the repository at this point in the history

  3. [Meta Schedule] Per-Store-Feature (#521)

    @junrushao @zxybazh
    junrushao authored and zxybazh committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    44957fc View commit details
    Browse the repository at this point in the history

  4. [Meta Schedule] Add XGBoost Model & Random Model (#519) 

    * Squashed commit

[Meta Schedule][M3c] Schedule Rules, Mutator & Postprocs (#485)

[Meta Schedule][M3c] PostOrderApply (#486)

Fix Post Order Apply (#490)

[MetaSchedule] Relay Integration (#489)

[M3c][Meta Schedule] Add Trace Correctness Test for PostOrderApply (#492)

Fix replay trace. (#493)

[M3c][Meta Schedule] Implement the Replay Func class. (#495)

[PR] Test script for meta-schedule task extraction. Interface to load… (#494)

[Meta Schedule Refactor] Get child blocks (#500)

Read-at && Write-at (#497)

[M3c][Meta Schedule] Measure Callbacks (#498)

[Bug] Fix Infinite Loop Caused When Calling Methods Not Overrided In PyClass (#496)

[MetaSchedule] Sample-Perfect-Tile (#501)

[MetaSchedule] TE Workloads (#502)

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Sunghyun Park <49998730+sunggg@users.noreply.github.com>

[TensorIR] GetProducer, GetConsumer (#506)

[MetaScheduleRefactor] Annotate&Unannotate (#505)

* annotate

* annotate

* lint

* test

* fix

* fix

* fix

[MetaSchedule] Rewrite Cooperative-Fetching / Unbound-Block / Reduction-Block (#509)

Fix sttr func & schedule naming.

Fix schedule -> sch.

Add feature extractor.

Fix init.

Add cost model.

Remove unused include.

[MetaSchedule] Rewrite Parallel-Vectorize-Unroll / Verify-GPU / Disallow-Dynamic-Loops (#499)

* wip

fix

* revoke change to gallery

* split postprocessors to separate files

* rename attrs

* minor

* minor tweak on utils.h

* refactor disallow-dynamic-loop

* refactor verify_gpu_code

* succesfully give up refactoring parallelize-vectorize-unroll

* python structuring

* unittests

Co-authored-by: Junru Shao <junrushao1994@gmail.com>

Fix issues.

Fix init.

Finish random model part.

Finish xgb model.

Minor fix.

Rebase.

Add init.

Await refactor of callback.

Update a bit on the test case.

Move impos.

Minor fix.

More fixes.

Remove unused import.

Fix per store feature test.

Update model save / load.

* Fix model save / load with tar.

* Fix issues.

* Remove dup.

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
    @zxybazh @junrushao
    zxybazh and junrushao committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    9f17b12 View commit details
    Browse the repository at this point in the history

  5. User-Interface: Tune-TIR (#525) 

    * User-Interface: Tune-TIR

* fix fix fix
    @junrushao @zxybazh
    junrushao authored and zxybazh committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    c587698 View commit details
    Browse the repository at this point in the history

  6. User-Interface: Tune-TE (#527) 

    * fix a lot of issues

* Add tune-te
    @junrushao @zxybazh
    junrushao authored and zxybazh committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    065a337 View commit details
    Browse the repository at this point in the history

  7. Get CUDA tuning working (#529)

    @junrushao @zxybazh
    junrushao authored and zxybazh committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    a028627 View commit details
    Browse the repository at this point in the history

  8. [Meta Schedule] Evolutionary Search (#522) 

    * Checkpoint.

Fix cost model comment.

Finish evolutionary seaarch.

Remove  extra code.

Fix compile.

Add comments.

Add python part.

Ad test.

Update other files & comments.

* Squashed commit

[Meta Schedule][M3c] Schedule Rules, Mutator & Postprocs (#485)

[Meta Schedule][M3c] PostOrderApply (#486)

Fix Post Order Apply (#490)

[MetaSchedule] Relay Integration (#489)

[M3c][Meta Schedule] Add Trace Correctness Test for PostOrderApply (#492)

Fix replay trace. (#493)

[M3c][Meta Schedule] Implement the Replay Func class. (#495)

[PR] Test script for meta-schedule task extraction. Interface to load… (#494)

[Meta Schedule Refactor] Get child blocks (#500)

Read-at && Write-at (#497)

[M3c][Meta Schedule] Measure Callbacks (#498)

[Bug] Fix Infinite Loop Caused When Calling Methods Not Overrided In PyClass (#496)

[MetaSchedule] Sample-Perfect-Tile (#501)

[MetaSchedule] TE Workloads (#502)

Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Sunghyun Park <49998730+sunggg@users.noreply.github.com>

* [TensorIR] GetProducer, GetConsumer (#506)

* [MetaScheduleRefactor] Annotate&Unannotate (#505)

* annotate

* annotate

* lint

* test

* fix

* fix

* fix

* [MetaSchedule] Rewrite Cooperative-Fetching / Unbound-Block / Reduction-Block (#509)

* Blockize & Tensorize (#514)

* Blockize & Tensorize

* Update tensor intrin

* Fix blockized & Recalculate affine flags

* Cleanup utils.cc

* Add test cases of blockize

* Re-enable affine flag checking

* Checkpoint.

Fix cost model comment.

Finish evolutionary seaarch.

Remove  extra code.

Fix compile.

Add comments.

Add python part.

Ad test.

Update other files & comments.

Fix random seed bug.

Minor fix.

Fix num-cores.

Add docs.

Check point.

Add max_fail_cnt.

Minor fix.

Minor fix.

Segfault.

Fix pointers to trace.

Test fix.

Remove measure callbacks.

Refactor a bit.

Split function.

Adjust variable name.

Minor fixes.

Add mutator probs to TuneContext.

Add token.

Fix loops.

Remove include.

Add has workload for database.

Add check.

Add concurrent bitmask.

* Fix TuneContext.

* Fix haash & stuff.

* Modifyy shash.

* Remove trace field.

* Minor fix.

* Fix cbmask.

* Fix numbers.

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
Co-authored-by: Siyuan Feng <Hzfengsy@sjtu.edu.cn>
Co-authored-by: Bohan Hou <32121147+spectrometerHBH@users.noreply.github.com>
Co-authored-by: Hongyi Jin <3231950289@qq.com>
Co-authored-by: Ruihang Lai <lairuihangdongdong@qq.com>
Co-authored-by: Wuwei Lin <wuwei@apache.org>
Co-authored-by: Sunghyun Park <49998730+sunggg@users.noreply.github.com>
    @zxybazh @junrushao
    @Hzfengsy @spectrometerHBH @jinhongyii @MasterJH5574 @vinx13 @sunggg
    8 people committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    77a8cab View commit details
    Browse the repository at this point in the history

  9. [BugFix] Remove duplicated definition of MakeMultinomialSampler (#535)

    @junrushao @zxybazh
    junrushao authored and zxybazh committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    a9ce77f View commit details
    Browse the repository at this point in the history

  10. Tune relay.

    @zxybazh
    zxybazh committed Dec 9, 2021
    Configuration menu
    Copy the full SHA
    c7c007d View commit details
    Browse the repository at this point in the history

Commits on Dec 10, 2021

  1. [FIX] Simplify during create prim func (#9691)

    @Hzfengsy @zxybazh
    Hzfengsy authored and zxybazh committed Dec 10, 2021
    Configuration menu
    Copy the full SHA
    51817c8 View commit details
    Browse the repository at this point in the history

  2. Further add interface.

    @zxybazh
    zxybazh committed Dec 10, 2021
    Configuration menu
    Copy the full SHA
    5049283 View commit details
    Browse the repository at this point in the history

  3. Remove unused import

    @zxybazh
    zxybazh committed Dec 10, 2021
    Configuration menu
    Copy the full SHA
    148f01b View commit details
    Browse the repository at this point in the history

  4. Fix rebase.

    @zxybazh
    zxybazh committed Dec 10, 2021
    Configuration menu
    Copy the full SHA
    5073a38 View commit details
    Browse the repository at this point in the history

  5. Renmove unused module.

    @zxybazh
    zxybazh committed Dec 10, 2021
    Configuration menu
    Copy the full SHA
    3be1388 View commit details
    Browse the repository at this point in the history