├── .gitignore ├── .pre-commit-config.yaml ├── CODE_OF_CONDUCT.md ├── LICENSE ├── MANIFEST.in ├── doc ├── CONTRIBUTING.md ├── bugs.md ├── cli.md ├── concept.md ├── known-issues.md └── log-and-err.md ├── env.sh ├── equality_saturation_helper ├── bug_code_generation.py ├── data_format.py ├── fuzz_loop.py ├── helper.py ├── minimizer.py └── rust_helper │ ├── Cargo.lock │ ├── Cargo.toml │ └── src │ ├── config.rs │ ├── graph.rs │ ├── lib.rs │ ├── model.rs │ └── rules.rs ├── exp_helper ├── 0_download_raw_data.sh ├── 1_sumarize_buglist_raw.py ├── 2_run_cmp_exp.py ├── 2_sumarize_cmp_exp_raw.py ├── 3_eq_effectiveness_raw.py ├── 3_run_eq_effectiveness.py ├── 4_run_throughput.py ├── 4_throughput_raw.py ├── bug_analysis.py ├── evaluate_tvm_log.py ├── process_diff_effect.py ├── test_env.py └── tvm_log_process.py ├── 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 └── viz_merged_cov.py ├── libdl_compiler_fuzzer_helper.so ├── nnsmith ├── __init__.py ├── abstract │ ├── __init__.py │ ├── arith.py │ ├── dtype.py │ ├── extension.py │ ├── op.py │ └── tensor.py ├── backends │ ├── __init__.py │ ├── factory.py │ ├── hidet.py │ ├── infinitensor.py │ ├── onnxruntime.py │ ├── pt2.py │ ├── tensorrt.py │ ├── tflite.py │ ├── torchjit.py │ ├── tvm.py │ └── xla.py ├── cli │ ├── __init__.py │ ├── ablation_exp_run.py │ ├── dtype_test.py │ ├── equivalent_fuzz.py │ ├── fuzz.py │ ├── model_exec.py │ ├── model_gen.py │ ├── run_a_testcase.py │ └── test_eq_helper.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 ├── readme.md ├── 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 /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/LICENSE -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /doc/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/doc/CONTRIBUTING.md -------------------------------------------------------------------------------- /doc/bugs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/doc/bugs.md -------------------------------------------------------------------------------- /doc/cli.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/doc/cli.md -------------------------------------------------------------------------------- /doc/concept.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/doc/concept.md -------------------------------------------------------------------------------- /doc/known-issues.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/doc/known-issues.md -------------------------------------------------------------------------------- /doc/log-and-err.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/doc/log-and-err.md -------------------------------------------------------------------------------- /env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/env.sh -------------------------------------------------------------------------------- /equality_saturation_helper/bug_code_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/bug_code_generation.py -------------------------------------------------------------------------------- /equality_saturation_helper/data_format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/data_format.py -------------------------------------------------------------------------------- /equality_saturation_helper/fuzz_loop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/fuzz_loop.py -------------------------------------------------------------------------------- /equality_saturation_helper/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/helper.py -------------------------------------------------------------------------------- /equality_saturation_helper/minimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/minimizer.py -------------------------------------------------------------------------------- /equality_saturation_helper/rust_helper/Cargo.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/rust_helper/Cargo.lock -------------------------------------------------------------------------------- /equality_saturation_helper/rust_helper/Cargo.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/rust_helper/Cargo.toml -------------------------------------------------------------------------------- /equality_saturation_helper/rust_helper/src/config.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/rust_helper/src/config.rs -------------------------------------------------------------------------------- /equality_saturation_helper/rust_helper/src/graph.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/rust_helper/src/graph.rs -------------------------------------------------------------------------------- /equality_saturation_helper/rust_helper/src/lib.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/rust_helper/src/lib.rs -------------------------------------------------------------------------------- /equality_saturation_helper/rust_helper/src/model.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/rust_helper/src/model.rs -------------------------------------------------------------------------------- /equality_saturation_helper/rust_helper/src/rules.rs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/equality_saturation_helper/rust_helper/src/rules.rs -------------------------------------------------------------------------------- /exp_helper/0_download_raw_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/0_download_raw_data.sh -------------------------------------------------------------------------------- /exp_helper/1_sumarize_buglist_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/1_sumarize_buglist_raw.py -------------------------------------------------------------------------------- /exp_helper/2_run_cmp_exp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/2_run_cmp_exp.py -------------------------------------------------------------------------------- /exp_helper/2_sumarize_cmp_exp_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/2_sumarize_cmp_exp_raw.py -------------------------------------------------------------------------------- /exp_helper/3_eq_effectiveness_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/3_eq_effectiveness_raw.py -------------------------------------------------------------------------------- /exp_helper/3_run_eq_effectiveness.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/3_run_eq_effectiveness.py -------------------------------------------------------------------------------- /exp_helper/4_run_throughput.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/4_run_throughput.py -------------------------------------------------------------------------------- /exp_helper/4_throughput_raw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/4_throughput_raw.py -------------------------------------------------------------------------------- /exp_helper/bug_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/bug_analysis.py -------------------------------------------------------------------------------- /exp_helper/evaluate_tvm_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/evaluate_tvm_log.py -------------------------------------------------------------------------------- /exp_helper/process_diff_effect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/process_diff_effect.py -------------------------------------------------------------------------------- /exp_helper/test_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/test_env.py -------------------------------------------------------------------------------- /exp_helper/tvm_log_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/exp_helper/tvm_log_process.py -------------------------------------------------------------------------------- /experiments/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/README.md -------------------------------------------------------------------------------- /experiments/evaluate_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/evaluate_models.py -------------------------------------------------------------------------------- /experiments/legacy/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/README.md -------------------------------------------------------------------------------- /experiments/legacy/batch_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/batch_eval.py -------------------------------------------------------------------------------- /experiments/legacy/cnt_uniq_crash.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/cnt_uniq_crash.py -------------------------------------------------------------------------------- /experiments/legacy/compare_tzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/compare_tzer.py -------------------------------------------------------------------------------- /experiments/legacy/cov_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/cov_eval.py -------------------------------------------------------------------------------- /experiments/legacy/cov_exp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/cov_exp.sh -------------------------------------------------------------------------------- /experiments/legacy/cov_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/cov_merge.py -------------------------------------------------------------------------------- /experiments/legacy/get_gentime_csv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/get_gentime_csv.py -------------------------------------------------------------------------------- /experiments/legacy/graphfuzz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/graphfuzz.py -------------------------------------------------------------------------------- /experiments/legacy/input_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/input_search.py -------------------------------------------------------------------------------- /experiments/legacy/input_search_exp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/input_search_exp.sh -------------------------------------------------------------------------------- /experiments/legacy/invalid_rate_torch_init.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/invalid_rate_torch_init.py -------------------------------------------------------------------------------- /experiments/legacy/lemon_tf2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/lemon_tf2onnx.py -------------------------------------------------------------------------------- /experiments/legacy/nnsmith_gen_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/nnsmith_gen_onnx.py -------------------------------------------------------------------------------- /experiments/legacy/onnx_graph_analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/onnx_graph_analyzer.py -------------------------------------------------------------------------------- /experiments/legacy/onnx_param_analyzer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/onnx_param_analyzer.py -------------------------------------------------------------------------------- /experiments/legacy/plot_cov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/plot_cov.py -------------------------------------------------------------------------------- /experiments/legacy/plot_inp_search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/plot_inp_search.py -------------------------------------------------------------------------------- /experiments/legacy/plot_inp_search_merge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/legacy/plot_inp_search_merge.py -------------------------------------------------------------------------------- /experiments/process_profraws.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/process_profraws.py -------------------------------------------------------------------------------- /experiments/viz_merged_cov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/experiments/viz_merged_cov.py -------------------------------------------------------------------------------- /libdl_compiler_fuzzer_helper.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/libdl_compiler_fuzzer_helper.so -------------------------------------------------------------------------------- /nnsmith/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/__init__.py -------------------------------------------------------------------------------- /nnsmith/abstract/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/abstract/__init__.py -------------------------------------------------------------------------------- /nnsmith/abstract/arith.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/abstract/arith.py -------------------------------------------------------------------------------- /nnsmith/abstract/dtype.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/abstract/dtype.py -------------------------------------------------------------------------------- /nnsmith/abstract/extension.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/abstract/extension.py -------------------------------------------------------------------------------- /nnsmith/abstract/op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/abstract/op.py -------------------------------------------------------------------------------- /nnsmith/abstract/tensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/abstract/tensor.py -------------------------------------------------------------------------------- /nnsmith/backends/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/__init__.py -------------------------------------------------------------------------------- /nnsmith/backends/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/factory.py -------------------------------------------------------------------------------- /nnsmith/backends/hidet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/hidet.py -------------------------------------------------------------------------------- /nnsmith/backends/infinitensor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/infinitensor.py -------------------------------------------------------------------------------- /nnsmith/backends/onnxruntime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/onnxruntime.py -------------------------------------------------------------------------------- /nnsmith/backends/pt2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/pt2.py -------------------------------------------------------------------------------- /nnsmith/backends/tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/tensorrt.py -------------------------------------------------------------------------------- /nnsmith/backends/tflite.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/tflite.py -------------------------------------------------------------------------------- /nnsmith/backends/torchjit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/torchjit.py -------------------------------------------------------------------------------- /nnsmith/backends/tvm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/tvm.py -------------------------------------------------------------------------------- /nnsmith/backends/xla.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/backends/xla.py -------------------------------------------------------------------------------- /nnsmith/cli/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nnsmith/cli/ablation_exp_run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/cli/ablation_exp_run.py -------------------------------------------------------------------------------- /nnsmith/cli/dtype_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/cli/dtype_test.py -------------------------------------------------------------------------------- /nnsmith/cli/equivalent_fuzz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/cli/equivalent_fuzz.py -------------------------------------------------------------------------------- /nnsmith/cli/fuzz.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/cli/fuzz.py -------------------------------------------------------------------------------- /nnsmith/cli/model_exec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/cli/model_exec.py -------------------------------------------------------------------------------- /nnsmith/cli/model_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/cli/model_gen.py -------------------------------------------------------------------------------- /nnsmith/cli/run_a_testcase.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/cli/run_a_testcase.py -------------------------------------------------------------------------------- /nnsmith/cli/test_eq_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/cli/test_eq_helper.py -------------------------------------------------------------------------------- /nnsmith/config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /nnsmith/config/hydra/job_logging/console.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/config/hydra/job_logging/console.yaml -------------------------------------------------------------------------------- /nnsmith/config/hydra/job_logging/file.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/config/hydra/job_logging/file.yaml -------------------------------------------------------------------------------- /nnsmith/config/main.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/config/main.yaml -------------------------------------------------------------------------------- /nnsmith/difftest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/difftest.py -------------------------------------------------------------------------------- /nnsmith/error.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/error.py -------------------------------------------------------------------------------- /nnsmith/filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/filter.py -------------------------------------------------------------------------------- /nnsmith/gir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/gir.py -------------------------------------------------------------------------------- /nnsmith/graph_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/graph_gen.py -------------------------------------------------------------------------------- /nnsmith/logging.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/logging.py -------------------------------------------------------------------------------- /nnsmith/macro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/macro.py -------------------------------------------------------------------------------- /nnsmith/materialize/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/__init__.py -------------------------------------------------------------------------------- /nnsmith/materialize/onnx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/onnx/__init__.py -------------------------------------------------------------------------------- /nnsmith/materialize/tensorflow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/tensorflow/__init__.py -------------------------------------------------------------------------------- /nnsmith/materialize/tensorflow/dialect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/tensorflow/dialect.py -------------------------------------------------------------------------------- /nnsmith/materialize/tensorflow/forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/tensorflow/forward.py -------------------------------------------------------------------------------- /nnsmith/materialize/tensorflow/tfnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/tensorflow/tfnet.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/torch/__init__.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/dialect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/torch/dialect.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/forward.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/torch/forward.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/input_gen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/torch/input_gen.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/numeric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/torch/numeric.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/torch/parse.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/proxy_grad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/torch/proxy_grad.py -------------------------------------------------------------------------------- /nnsmith/materialize/torch/symbolnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/materialize/torch/symbolnet.py -------------------------------------------------------------------------------- /nnsmith/narrow_spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/narrow_spec.py -------------------------------------------------------------------------------- /nnsmith/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/nnsmith/util.py -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/pyproject.toml -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/readme.md -------------------------------------------------------------------------------- /requirements/core.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/requirements/core.txt -------------------------------------------------------------------------------- /requirements/dev.txt: -------------------------------------------------------------------------------- 1 | pytest 2 | pre-commit 3 | black 4 | wget 5 | gputil 6 | pygraphviz 7 | -------------------------------------------------------------------------------- /requirements/exp.txt: -------------------------------------------------------------------------------- 1 | pandas==1.3.5 2 | tqdm 3 | -------------------------------------------------------------------------------- /requirements/install_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/requirements/install_all.sh -------------------------------------------------------------------------------- /requirements/sys/onnx.txt: -------------------------------------------------------------------------------- 1 | onnx 2 | -------------------------------------------------------------------------------- /requirements/sys/onnxruntime.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/requirements/sys/onnxruntime.txt -------------------------------------------------------------------------------- /requirements/sys/tensorflow.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/requirements/sys/tensorflow.txt -------------------------------------------------------------------------------- /requirements/sys/tensorrt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/requirements/sys/tensorrt.txt -------------------------------------------------------------------------------- /requirements/sys/torch.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/requirements/sys/torch.txt -------------------------------------------------------------------------------- /requirements/sys/tvm.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/requirements/sys/tvm.txt -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/setup.cfg -------------------------------------------------------------------------------- /tests/core/test_extra_constraint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/core/test_extra_constraint.py -------------------------------------------------------------------------------- /tests/core/test_gir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/core/test_gir.py -------------------------------------------------------------------------------- /tests/core/test_native_pickable.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/core/test_native_pickable.py -------------------------------------------------------------------------------- /tests/core/test_parse_name_kwargs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/core/test_parse_name_kwargs.py -------------------------------------------------------------------------------- /tests/mock/filter_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/mock/filter_patch.py -------------------------------------------------------------------------------- /tests/mock/requires_patch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/mock/requires_patch.py -------------------------------------------------------------------------------- /tests/onnxruntime/test_ort_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/onnxruntime/test_ort_backend.py -------------------------------------------------------------------------------- /tests/tensorflow/test_dump_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/tensorflow/test_dump_load.py -------------------------------------------------------------------------------- /tests/tensorflow/test_tflite_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/tensorflow/test_tflite_backend.py -------------------------------------------------------------------------------- /tests/tensorflow/test_xla_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/tensorflow/test_xla_backend.py -------------------------------------------------------------------------------- /tests/tensorrt/test_trt_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/tensorrt/test_trt_backend.py -------------------------------------------------------------------------------- /tests/torch/test_biconvert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/torch/test_biconvert.py -------------------------------------------------------------------------------- /tests/torch/test_dump_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/torch/test_dump_load.py -------------------------------------------------------------------------------- /tests/torch/test_pt2_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/torch/test_pt2_backend.py -------------------------------------------------------------------------------- /tests/torch/test_render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/torch/test_render.py -------------------------------------------------------------------------------- /tests/torch/test_torchjit_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/torch/test_torchjit_backend.py -------------------------------------------------------------------------------- /tests/tvm/test_tvm_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ChijinZ/PolyJuice-Fuzzer/HEAD/tests/tvm/test_tvm_backend.py --------------------------------------------------------------------------------