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.

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

[Relay][Pass] Add ExtractOperators pass #8996

Merged
merged 11 commits into from
Sep 16, 2021
Merged

Conversation

anwang2009
Copy link
Contributor

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

tests/python/relay/test_analysis_extract_operators.py Outdated Show resolved Hide resolved
src/relay/analysis/extract_operators.cc Outdated Show resolved Hide resolved
src/relay/analysis/extract_operators.cc Show resolved Hide resolved
python/tvm/relay/analysis/analysis.py Outdated Show resolved Hide resolved
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
@jroesch
Copy link
Member

jroesch commented Sep 13, 2021

lgtm, will let @comaniac own this one.

Copy link
Contributor

@comaniac comaniac left a comment

Choose a reason for hiding this comment

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

LGTM. Just nits.

src/relay/analysis/extract_operators.cc Outdated Show resolved Hide resolved
python/tvm/relay/analysis/analysis.py Outdated Show resolved Hide resolved
anwang2009 and others added 2 commits September 13, 2021 15:48
Co-authored-by: Cody Yu <comaniac0422@gmail.com>
@masahi masahi merged commit 777ace3 into apache:main Sep 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 free to 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