Skip to content

[Relay][Pass] Add ExtractOperators pass - #8996

Merged
masahi merged 11 commits into
apache:mainfrom
anwang2009:aw/operator-pass
Sep 16, 2021
Merged

[Relay][Pass] Add ExtractOperators pass#8996
masahi merged 11 commits into
apache:mainfrom
anwang2009:aw/operator-pass

Conversation

@anwang2009

Copy link
Copy Markdown
Contributor

Adds helper pass to extract unique operator names from a module.

Comment threadtests/python/relay/test_analysis_extract_operators.py Outdated
Comment threadsrc/relay/analysis/extract_operators.cc Outdated
Comment threadsrc/relay/analysis/extract_operators.cc
Comment threadpython/tvm/relay/analysis/analysis.py Outdated
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
@jroesch

Copy link
Copy Markdown
Member

lgtm, will let @comaniac own this one.

@comaniaccomaniac left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM. Just nits.

Comment threadsrc/relay/analysis/extract_operators.cc Outdated
Comment threadpython/tvm/relay/analysis/analysis.py Outdated
anwang2009and others added 2 commits September 13, 2021 15:48
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
Comment threadsrc/relay/analysis/extract_operators.cc Outdated
Comment threadsrc/relay/analysis/extract_operators.cc Outdated
@masahi
masahi merged commit 777ace3 into apache:mainSep 16, 2021
AndrewZhaoLuo added a commit to AndrewZhaoLuo/tvm that referenced this pull request Sep 16, 2021
* main: (102 commits)
Implementation of relay_to_tir target hook (apache#8423)
[Onnx] Fix NLL Loss tests (apache#8971)
[Bugfix] Fix other div zero errors also in rewrite_simplify (apache#8983)
[ONNX] enable the onnx tests after PR apache#8274 merged (apache#9019)
[Hexagon] Disable `thread_local` on Hexagon (apache#9025)
[Hexagon] Allow undefined symbols in libtvm_runtime.so on Hexagon (apache#9024)
[Onnx] Add momentum (apache#9000)
fix (apache#9021)
[Community] @AndrewZhaoLuo -> Reviewer (apache#9020)
[Hexagon] Implement model launcher (apache#8986)
[Relay][Pass] Add ExtractOperators pass (apache#8996)
[BYOC][TensorRT] Add TensorRT own int8 calibration support to TensorRT BYOC integration (apache#8808)
[ONNX] Add Einsum converter (apache#8985)
Add standalone_crt/ to be part of the wheel package, when available. (apache#9005)
[Relay] Remove memory planing from LowerTEPass (apache#8974)
[Hexagon] Treat floats as float32 when passing args to offloaded kernels (apache#9010)
[Runtime] Pipeline Executor Initial patch. (apache#8702)
[Hexagon] `llvm-options` attribute is an array of strings (apache#9011)
disable cuda int8 schedule for non-cuda gpu target (apache#9014)
[Torch] Add an option to make imported models compatible with the Relay text parser (apache#9015)
...
ylc pushed a commit to ylc/tvm that referenced this pull request Sep 29, 2021
* add extractor
* extract to array
* add comments
* lint
* Update tests/python/relay/test_analysis_extract_operators.py
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
* op freqs
* add comment
* Update python/tvm/relay/analysis/analysis.py
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
* oops
* mixedmode visitor
* oops
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
ylc pushed a commit to ylc/tvm that referenced this pull request Jan 13, 2022
* add extractor
* extract to array
* add comments
* lint
* Update tests/python/relay/test_analysis_extract_operators.py
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
* op freqs
* add comment
* Update python/tvm/relay/analysis/analysis.py
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
* oops
* mixedmode visitor
* oops
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
Sign up for freeto join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants

@anwang2009@jroesch@masahi@comaniac@AndrewZhaoLuo