├── .github ├── ISSUE_TEMPLATE │ ├── bug-report.yml │ ├── config.yml │ └── feature-request.yml ├── actions │ └── setup │ │ └── action.yaml └── workflows │ ├── ci.yaml │ └── pre-commit.yaml ├── .gitignore ├── .pre-commit-config.yaml ├── CODE_OF_CONDUCT.md ├── LICENSE ├── MANIFEST.in ├── README.md ├── doc ├── CONTRIBUTING.md ├── bug_summary.py ├── bugs.md ├── cli.md ├── concept.md ├── known-issues.md └── log-and-err.md ├── experiments ├── README.md ├── evaluate_models.py ├── legacy │ ├── README.md │ ├── batch_eval.py │ ├── cnt_uniq_crash.py │ ├── compare_tzer.py │ ├── cov_eval.py │ ├── cov_exp.sh │ ├── cov_merge.py │ ├── get_gentime_csv.py │ ├── graphfuzz.py │ ├── input_search.py │ ├── input_search_exp.sh │ ├── invalid_rate_torch_init.py │ ├── lemon_tf2onnx.py │ ├── nnsmith_gen_onnx.py │ ├── onnx_graph_analyzer.py │ ├── onnx_param_analyzer.py │ ├── plot_cov.py │ ├── plot_inp_search.py │ └── plot_inp_search_merge.py ├── process_profraws.py ├── requirements.txt └── viz_merged_cov.py ├── nnsmith ├── __init__.py ├── abstract │ ├── __init__.py │ ├── arith.py │ ├── dtype.py │ ├── extension.py │ ├── op.py │ └── tensor.py ├── backends │ ├── __init__.py │ ├── factory.py │ ├── onnxruntime.py │ ├── pt2.py │ ├── tensorrt.py │ ├── tflite.py │ ├── torchjit.py │ ├── tvm.py │ └── xla.py ├── cli │ ├── __init__.py │ ├── dtype_test.py │ ├── fuzz.py │ ├── model_exec.py │ ├── model_gen.py │ └── report_syn.py ├── config │ ├── __init__.py │ ├── hydra │ │ └── job_logging │ │ │ ├── console.yaml │ │ │ └── file.yaml │ └── main.yaml ├── difftest.py ├── error.py ├── filter.py ├── gir.py ├── graph_gen.py ├── logging.py ├── macro.py ├── materialize │ ├── __init__.py │ ├── onnx │ │ └── __init__.py │ ├── tensorflow │ │ ├── __init__.py │ │ ├── dialect.py │ │ ├── forward.py │ │ └── tfnet.py │ └── torch │ │ ├── __init__.py │ │ ├── dialect.py │ │ ├── forward.py │ │ ├── input_gen.py │ │ ├── numeric.py │ │ ├── parse.py │ │ ├── proxy_grad.py │ │ └── symbolnet.py ├── narrow_spec.py └── util.py ├── pyproject.toml ├── requirements ├── core.txt ├── dev.txt ├── exp.txt ├── install_all.sh └── sys │ ├── onnx.txt │ ├── onnxruntime.txt │ ├── tensorflow.txt │ ├── tensorrt.txt │ ├── torch.txt │ └── tvm.txt ├── setup.cfg └── tests ├── core ├── test_extra_constraint.py ├── test_gir.py ├── test_native_pickable.py └── test_parse_name_kwargs.py ├── mock ├── filter_patch.py └── requires_patch.py ├── onnxruntime └── test_ort_backend.py ├── tensorflow ├── test_dump_load.py ├── test_tflite_backend.py └── test_xla_backend.py ├── tensorrt └── test_trt_backend.py ├── torch ├── test_biconvert.py ├── test_dump_load.py ├── test_pt2_backend.py ├── test_render.py └── test_torchjit_backend.py └── tvm └── test_tvm_backend.py /.github/ISSUE_TEMPLATE/bug-report.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/.github/ISSUE_TEMPLATE/bug-report.yml -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/config.yml: -------------------------------------------------------------------------------- 1 | blank_issues_enabled: true 2 | -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature-request.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/.github/ISSUE_TEMPLATE/feature-request.yml -------------------------------------------------------------------------------- /.github/actions/setup/action.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/.github/actions/setup/action.yaml -------------------------------------------------------------------------------- /.github/workflows/ci.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/.github/workflows/ci.yaml -------------------------------------------------------------------------------- /.github/workflows/pre-commit.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/.github/workflows/pre-commit.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/README.md -------------------------------------------------------------------------------- /doc/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/doc/CONTRIBUTING.md -------------------------------------------------------------------------------- /doc/bug_summary.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/doc/bug_summary.py -------------------------------------------------------------------------------- /doc/bugs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/doc/bugs.md -------------------------------------------------------------------------------- /doc/cli.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/doc/cli.md -------------------------------------------------------------------------------- /doc/concept.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/doc/concept.md -------------------------------------------------------------------------------- /doc/known-issues.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/doc/known-issues.md -------------------------------------------------------------------------------- /doc/log-and-err.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/doc/log-and-err.md -------------------------------------------------------------------------------- /experiments/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/README.md -------------------------------------------------------------------------------- /experiments/evaluate_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/evaluate_models.py -------------------------------------------------------------------------------- /experiments/legacy/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/README.md -------------------------------------------------------------------------------- /experiments/legacy/batch_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/batch_eval.py -------------------------------------------------------------------------------- /experiments/legacy/cnt_uniq_crash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/cnt_uniq_crash.py -------------------------------------------------------------------------------- /experiments/legacy/compare_tzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/compare_tzer.py -------------------------------------------------------------------------------- /experiments/legacy/cov_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/cov_eval.py -------------------------------------------------------------------------------- /experiments/legacy/cov_exp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/cov_exp.sh -------------------------------------------------------------------------------- /experiments/legacy/cov_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/cov_merge.py -------------------------------------------------------------------------------- /experiments/legacy/get_gentime_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/get_gentime_csv.py -------------------------------------------------------------------------------- /experiments/legacy/graphfuzz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/graphfuzz.py -------------------------------------------------------------------------------- /experiments/legacy/input_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/input_search.py -------------------------------------------------------------------------------- /experiments/legacy/input_search_exp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/input_search_exp.sh -------------------------------------------------------------------------------- /experiments/legacy/invalid_rate_torch_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/invalid_rate_torch_init.py -------------------------------------------------------------------------------- /experiments/legacy/lemon_tf2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/lemon_tf2onnx.py -------------------------------------------------------------------------------- /experiments/legacy/nnsmith_gen_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/nnsmith_gen_onnx.py -------------------------------------------------------------------------------- /experiments/legacy/onnx_graph_analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/onnx_graph_analyzer.py -------------------------------------------------------------------------------- /experiments/legacy/onnx_param_analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/onnx_param_analyzer.py -------------------------------------------------------------------------------- /experiments/legacy/plot_cov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/plot_cov.py -------------------------------------------------------------------------------- /experiments/legacy/plot_inp_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/plot_inp_search.py -------------------------------------------------------------------------------- /experiments/legacy/plot_inp_search_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/legacy/plot_inp_search_merge.py -------------------------------------------------------------------------------- /experiments/process_profraws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/process_profraws.py -------------------------------------------------------------------------------- /experiments/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/requirements.txt -------------------------------------------------------------------------------- /experiments/viz_merged_cov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/experiments/viz_merged_cov.py -------------------------------------------------------------------------------- /nnsmith/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/__init__.py -------------------------------------------------------------------------------- /nnsmith/abstract/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/abstract/__init__.py -------------------------------------------------------------------------------- /nnsmith/abstract/arith.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/abstract/arith.py -------------------------------------------------------------------------------- /nnsmith/abstract/dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/abstract/dtype.py -------------------------------------------------------------------------------- /nnsmith/abstract/extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/abstract/extension.py -------------------------------------------------------------------------------- /nnsmith/abstract/op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/abstract/op.py -------------------------------------------------------------------------------- /nnsmith/abstract/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/abstract/tensor.py -------------------------------------------------------------------------------- /nnsmith/backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/backends/__init__.py -------------------------------------------------------------------------------- /nnsmith/backends/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/backends/factory.py -------------------------------------------------------------------------------- /nnsmith/backends/onnxruntime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/backends/onnxruntime.py -------------------------------------------------------------------------------- /nnsmith/backends/pt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/backends/pt2.py -------------------------------------------------------------------------------- /nnsmith/backends/tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/backends/tensorrt.py -------------------------------------------------------------------------------- /nnsmith/backends/tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/backends/tflite.py -------------------------------------------------------------------------------- /nnsmith/backends/torchjit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/backends/torchjit.py -------------------------------------------------------------------------------- /nnsmith/backends/tvm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/backends/tvm.py -------------------------------------------------------------------------------- /nnsmith/backends/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/backends/xla.py -------------------------------------------------------------------------------- /nnsmith/cli/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nnsmith/cli/dtype_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/cli/dtype_test.py -------------------------------------------------------------------------------- /nnsmith/cli/fuzz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/cli/fuzz.py -------------------------------------------------------------------------------- /nnsmith/cli/model_exec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/cli/model_exec.py -------------------------------------------------------------------------------- /nnsmith/cli/model_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/cli/model_gen.py -------------------------------------------------------------------------------- /nnsmith/cli/report_syn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/cli/report_syn.py -------------------------------------------------------------------------------- /nnsmith/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nnsmith/config/hydra/job_logging/console.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/config/hydra/job_logging/console.yaml -------------------------------------------------------------------------------- /nnsmith/config/hydra/job_logging/file.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/config/hydra/job_logging/file.yaml -------------------------------------------------------------------------------- /nnsmith/config/main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/config/main.yaml -------------------------------------------------------------------------------- /nnsmith/difftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/difftest.py -------------------------------------------------------------------------------- /nnsmith/error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/error.py -------------------------------------------------------------------------------- /nnsmith/filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/filter.py -------------------------------------------------------------------------------- /nnsmith/gir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/gir.py -------------------------------------------------------------------------------- /nnsmith/graph_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/graph_gen.py -------------------------------------------------------------------------------- /nnsmith/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/logging.py -------------------------------------------------------------------------------- /nnsmith/macro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/macro.py -------------------------------------------------------------------------------- /nnsmith/materialize/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/__init__.py -------------------------------------------------------------------------------- /nnsmith/materialize/onnx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/onnx/__init__.py -------------------------------------------------------------------------------- /nnsmith/materialize/tensorflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/tensorflow/__init__.py -------------------------------------------------------------------------------- /nnsmith/materialize/tensorflow/dialect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/tensorflow/dialect.py -------------------------------------------------------------------------------- /nnsmith/materialize/tensorflow/forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/tensorflow/forward.py -------------------------------------------------------------------------------- /nnsmith/materialize/tensorflow/tfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/tensorflow/tfnet.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/torch/__init__.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/dialect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/torch/dialect.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/torch/forward.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/input_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/torch/input_gen.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/numeric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/torch/numeric.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/torch/parse.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/proxy_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/torch/proxy_grad.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/symbolnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/materialize/torch/symbolnet.py -------------------------------------------------------------------------------- /nnsmith/narrow_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/narrow_spec.py -------------------------------------------------------------------------------- /nnsmith/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/nnsmith/util.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements/core.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/requirements/core.txt -------------------------------------------------------------------------------- /requirements/dev.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/requirements/dev.txt -------------------------------------------------------------------------------- /requirements/exp.txt: -------------------------------------------------------------------------------- 1 | pandas==1.3.5 2 | tqdm 3 | -------------------------------------------------------------------------------- /requirements/install_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/requirements/install_all.sh -------------------------------------------------------------------------------- /requirements/sys/onnx.txt: -------------------------------------------------------------------------------- 1 | onnx 2 | -------------------------------------------------------------------------------- /requirements/sys/onnxruntime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/requirements/sys/onnxruntime.txt -------------------------------------------------------------------------------- /requirements/sys/tensorflow.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/requirements/sys/tensorflow.txt -------------------------------------------------------------------------------- /requirements/sys/tensorrt.txt: -------------------------------------------------------------------------------- 1 | tensorrt 2 | -------------------------------------------------------------------------------- /requirements/sys/torch.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/requirements/sys/torch.txt -------------------------------------------------------------------------------- /requirements/sys/tvm.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/requirements/sys/tvm.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/setup.cfg -------------------------------------------------------------------------------- /tests/core/test_extra_constraint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/core/test_extra_constraint.py -------------------------------------------------------------------------------- /tests/core/test_gir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/core/test_gir.py -------------------------------------------------------------------------------- /tests/core/test_native_pickable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/core/test_native_pickable.py -------------------------------------------------------------------------------- /tests/core/test_parse_name_kwargs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/core/test_parse_name_kwargs.py -------------------------------------------------------------------------------- /tests/mock/filter_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/mock/filter_patch.py -------------------------------------------------------------------------------- /tests/mock/requires_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/mock/requires_patch.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_ort_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/onnxruntime/test_ort_backend.py -------------------------------------------------------------------------------- /tests/tensorflow/test_dump_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/tensorflow/test_dump_load.py -------------------------------------------------------------------------------- /tests/tensorflow/test_tflite_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/tensorflow/test_tflite_backend.py -------------------------------------------------------------------------------- /tests/tensorflow/test_xla_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/tensorflow/test_xla_backend.py -------------------------------------------------------------------------------- /tests/tensorrt/test_trt_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/tensorrt/test_trt_backend.py -------------------------------------------------------------------------------- /tests/torch/test_biconvert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/torch/test_biconvert.py -------------------------------------------------------------------------------- /tests/torch/test_dump_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/torch/test_dump_load.py -------------------------------------------------------------------------------- /tests/torch/test_pt2_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/torch/test_pt2_backend.py -------------------------------------------------------------------------------- /tests/torch/test_render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/torch/test_render.py -------------------------------------------------------------------------------- /tests/torch/test_torchjit_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/torch/test_torchjit_backend.py -------------------------------------------------------------------------------- /tests/tvm/test_tvm_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ise-uiuc/nnsmith/HEAD/tests/tvm/test_tvm_backend.py --------------------------------------------------------------------------------