Skip to content
This repository was archived by the owner on Sep 12, 2025. It is now read-only.

Refactor of nn-Meter project#41

Merged
Lynazhang merged 19 commits intomainfrom
nn-meter-refactor
Nov 9, 2021
Merged

Refactor of nn-Meter project#41
Lynazhang merged 19 commits intomainfrom
nn-meter-refactor

Conversation

@JiahangXu
Copy link
Copy Markdown
Collaborator

No description provided.

@JiahangXu JiahangXu marked this pull request as ready for review November 9, 2021 06:52
@JiahangXu JiahangXu requested a review from Lynazhang November 9, 2021 07:08
@JiahangXu
Copy link
Copy Markdown
Collaborator Author

JiahangXu commented Nov 9, 2021

Structure after refactor:

.
├── README.md
├── docs
│ ├── dataset.md
│ ├── imgs
│ ├── input_models.md
│ ├── kernel.md
│ ├── ops.md
│ ├── overview.md
│ ├── predictor
│ │ ├── hardware-aware-model-design.md
│ │ └── usage.md
│ ├── quick_start.md
│ └── requirements
│ └── requirements.txt
├── examples
│ ├── README.md
│ ├── nn-meter_dataset_for_gnn.ipynb
│ ├── nn-meter_predictor_for_bench_dataset.ipynb
│ └── nn-meter_predictor_for_different_model_format.ipynb
├── nn_meter
│ ├── init.py
│ ├── configs
│ │ └── predictors.yaml
│ ├── dataset
│ │ ├── init.py
│ │ ├── bench_dataset.py
│ │ └── gnn_dataloader.py
│ ├── ir_converter
│ │ ├── init.py
│ │ ├── frozenpb_converter
│ │ │ ├── init.py
│ │ │ ├── frozenpb_converter.py
│ │ │ ├── frozenpb_parser.py
│ │ │ ├── protobuf_helper.py
│ │ │ ├── shape_fetcher.py
│ │ │ └── shape_inference.py
│ │ ├── onnx_converter
│ │ │ ├── init.py
│ │ │ ├── constants.py
│ │ │ ├── converter.py
│ │ │ └── utils.py
│ │ ├── torch_converter
│ │ │ ├── init.py
│ │ │ ├── converter.py
│ │ │ └── opset_map.py
│ │ └── utils.py
│ ├── kernel_detector
│ │ ├── init.py
│ │ ├── fusionlib
│ │ │ ├── init.py
│ │ │ ├── add-relu_fusionunit.json
│ │ │ ├── bn-relu_fusionunit.json
│ │ │ ├── channelshuffle_fusionunit.json
│ │ │ ├── conv-bn-relu_fusionunit.json
│ │ │ ├── conv-bn_fusionunit.json
│ │ │ ├── dwconv-bn-relu_fusionunit.json
│ │ │ ├── elewise_fusionunit.json
│ │ │ ├── gap_fusionunit.json
│ │ │ ├── hswish_fusionunit.json
│ │ │ ├── se_fusionunit.json
│ │ │ └── utils.py
│ │ ├── kernel_detector.py
│ │ ├── rulelib
│ │ │ ├── init.py
│ │ │ ├── rule_reader.py
│ │ │ └── rule_splitter.py
│ │ └── utils
│ │ ├── init.py
│ │ ├── constants.py
│ │ ├── fusion_aware_graph.py
│ │ ├── ir_tools.py
│ │ ├── match_helper.py
│ │ └── union_find.py
│ ├── nn_meter_cli.py
│ ├── predictor
│ │ ├── init.py
│ │ ├── nn_meter_predictor.py
│ │ ├── prediction
│ │ │ ├── init.py
│ │ │ ├── extract_feature.py
│ │ │ ├── kernel_predictor.py
│ │ │ ├── predict_by_kernel.py
│ │ │ └── utils.py
│ │ └── utils.py
│ └── utils
│ ├── init.py
│ ├── config_manager.py
│ ├── graph_tool.py
│ ├── import_package.py
│ └── utils.py
├── setup.py
└── tests
├── README.md
├── data
│ ├── reference_result.txt
│ ├── reference_result_nni_based_torch.txt
│ ├── reference_result_onnx_based_torch.txt
│ ├── tensorflow_file.pb
│ └── torchmodels
├── integration_test.py
├── integration_test_torch.py
└── test_result.txt

@Lynazhang Lynazhang merged commit 759854e into main Nov 9, 2021
@JiahangXu JiahangXu deleted the nn-meter-refactor branch November 9, 2021 09:50
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants