├── .gitignore ├── README.md ├── autoscheduler ├── autoscheduler_log.py ├── autotvm_log.py ├── examine_task_extraction_failure.py ├── pass_context_not_preserved.py ├── tune_network_cuda_not_working_vm.py └── tune_network_llvm_test_graph_tuner.py ├── codegen ├── benchmark_recompile.py ├── test_export_to_ll.py └── tuning_records │ ├── tuning_records_fp16 │ └── tuning_records_fp32 ├── coreml_debug └── coreml-test.py ├── fp16_pass ├── README ├── benchmark_experiments │ ├── cuda │ │ ├── bert.py │ │ └── yolov2.py │ ├── llvm │ │ ├── bert.py │ │ └── yolov2.py │ └── metal │ │ ├── bert.py │ │ └── yolov2.py ├── benchmark_fp16.py ├── dump_bert.py ├── example_overwrite_settings.py ├── example_use_accumulators.py ├── integration_tests.py ├── test_upcasting_relay.py └── test_upcasting_tir.py ├── fq2i_pass ├── test_fq2i_pass.py └── test_tflite_to_onnx.py ├── metaschedule └── lower_relay_to_script.py ├── misc └── uint16_to_bfloat.py ├── octomizer └── rebenchmark_model.py ├── onnx ├── change_onnx_input_shapes.py ├── debug_caffenet_int8_import.py ├── export_bigly_model.py ├── export_resnet.py ├── onnx_change_batch_size.py ├── onnx_optimizer.py ├── quantize_models_example.py ├── test_onnx_profiler.py ├── test_slow_tune.py ├── test_split_onnx_graph.py └── test_split_onnx_graph_old.py ├── performance └── test_dense_packed.py ├── quantization ├── example_gather_lookup_table.py ├── issues_with_fq2i_pass.py ├── qnn_lower_example.py └── tanh_runge_katta.py ├── relay ├── 1x1_conv_to_gemm.py ├── debug_adam.py ├── debug_conv2d_transpose.py ├── debug_qnn_grouped_convs.py ├── depthwise_convolution_inconsistency.py ├── dynamic_to_static_example.py ├── example_fold_constant_pass.py ├── example_list_all_ops.py ├── graph_debugger_example.py ├── import_onnx_model.py ├── inputs.pkl ├── inputs2.pkl ├── inputs3.pkl ├── nhwc_layout_transform_example.py ├── test_batch_matmul.py ├── test_dropout.py ├── test_layout_transform_no_effect.py ├── test_layout_transform_pass.py ├── test_multiply_by_0_nan.py ├── test_multiply_by_0_nan_data │ ├── inputs.pkl │ ├── inputs2.pkl │ └── inputs3.pkl └── test_reshape_extraneous.py ├── resnet-18-NCHW-B1-llvm.json ├── resnet-18-NHWC-B1-llvm.json ├── runtime └── test_benchmark_funcs_graph_runtime.py ├── tir ├── lowering_to_tir_autoscheduler_from_logs.py ├── lowering_to_tir_autotvm_from_logs.py ├── te_matrix_multiplication.py ├── test_inconsistent_tir_lowering.py ├── test_inconsistent_tir_printing.py └── test_serialize_dag.py └── topi └── conv2d_cuda_exploration.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/README.md -------------------------------------------------------------------------------- /autoscheduler/autoscheduler_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/autoscheduler/autoscheduler_log.py -------------------------------------------------------------------------------- /autoscheduler/autotvm_log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/autoscheduler/autotvm_log.py -------------------------------------------------------------------------------- /autoscheduler/examine_task_extraction_failure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/autoscheduler/examine_task_extraction_failure.py -------------------------------------------------------------------------------- /autoscheduler/pass_context_not_preserved.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/autoscheduler/pass_context_not_preserved.py -------------------------------------------------------------------------------- /autoscheduler/tune_network_cuda_not_working_vm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/autoscheduler/tune_network_cuda_not_working_vm.py -------------------------------------------------------------------------------- /autoscheduler/tune_network_llvm_test_graph_tuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/autoscheduler/tune_network_llvm_test_graph_tuner.py -------------------------------------------------------------------------------- /codegen/benchmark_recompile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/codegen/benchmark_recompile.py -------------------------------------------------------------------------------- /codegen/test_export_to_ll.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/codegen/test_export_to_ll.py -------------------------------------------------------------------------------- /codegen/tuning_records/tuning_records_fp16: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/codegen/tuning_records/tuning_records_fp16 -------------------------------------------------------------------------------- /codegen/tuning_records/tuning_records_fp32: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/codegen/tuning_records/tuning_records_fp32 -------------------------------------------------------------------------------- /coreml_debug/coreml-test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/coreml_debug/coreml-test.py -------------------------------------------------------------------------------- /fp16_pass/README: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/README -------------------------------------------------------------------------------- /fp16_pass/benchmark_experiments/cuda/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/benchmark_experiments/cuda/bert.py -------------------------------------------------------------------------------- /fp16_pass/benchmark_experiments/cuda/yolov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/benchmark_experiments/cuda/yolov2.py -------------------------------------------------------------------------------- /fp16_pass/benchmark_experiments/llvm/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/benchmark_experiments/llvm/bert.py -------------------------------------------------------------------------------- /fp16_pass/benchmark_experiments/llvm/yolov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/benchmark_experiments/llvm/yolov2.py -------------------------------------------------------------------------------- /fp16_pass/benchmark_experiments/metal/bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/benchmark_experiments/metal/bert.py -------------------------------------------------------------------------------- /fp16_pass/benchmark_experiments/metal/yolov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/benchmark_experiments/metal/yolov2.py -------------------------------------------------------------------------------- /fp16_pass/benchmark_fp16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/benchmark_fp16.py -------------------------------------------------------------------------------- /fp16_pass/dump_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/dump_bert.py -------------------------------------------------------------------------------- /fp16_pass/example_overwrite_settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/example_overwrite_settings.py -------------------------------------------------------------------------------- /fp16_pass/example_use_accumulators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/example_use_accumulators.py -------------------------------------------------------------------------------- /fp16_pass/integration_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/integration_tests.py -------------------------------------------------------------------------------- /fp16_pass/test_upcasting_relay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/test_upcasting_relay.py -------------------------------------------------------------------------------- /fp16_pass/test_upcasting_tir.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fp16_pass/test_upcasting_tir.py -------------------------------------------------------------------------------- /fq2i_pass/test_fq2i_pass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fq2i_pass/test_fq2i_pass.py -------------------------------------------------------------------------------- /fq2i_pass/test_tflite_to_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/fq2i_pass/test_tflite_to_onnx.py -------------------------------------------------------------------------------- /metaschedule/lower_relay_to_script.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/metaschedule/lower_relay_to_script.py -------------------------------------------------------------------------------- /misc/uint16_to_bfloat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/misc/uint16_to_bfloat.py -------------------------------------------------------------------------------- /octomizer/rebenchmark_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/octomizer/rebenchmark_model.py -------------------------------------------------------------------------------- /onnx/change_onnx_input_shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/change_onnx_input_shapes.py -------------------------------------------------------------------------------- /onnx/debug_caffenet_int8_import.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/debug_caffenet_int8_import.py -------------------------------------------------------------------------------- /onnx/export_bigly_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/export_bigly_model.py -------------------------------------------------------------------------------- /onnx/export_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/export_resnet.py -------------------------------------------------------------------------------- /onnx/onnx_change_batch_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/onnx_change_batch_size.py -------------------------------------------------------------------------------- /onnx/onnx_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/onnx_optimizer.py -------------------------------------------------------------------------------- /onnx/quantize_models_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/quantize_models_example.py -------------------------------------------------------------------------------- /onnx/test_onnx_profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/test_onnx_profiler.py -------------------------------------------------------------------------------- /onnx/test_slow_tune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/test_slow_tune.py -------------------------------------------------------------------------------- /onnx/test_split_onnx_graph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/test_split_onnx_graph.py -------------------------------------------------------------------------------- /onnx/test_split_onnx_graph_old.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/onnx/test_split_onnx_graph_old.py -------------------------------------------------------------------------------- /performance/test_dense_packed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/performance/test_dense_packed.py -------------------------------------------------------------------------------- /quantization/example_gather_lookup_table.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/quantization/example_gather_lookup_table.py -------------------------------------------------------------------------------- /quantization/issues_with_fq2i_pass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/quantization/issues_with_fq2i_pass.py -------------------------------------------------------------------------------- /quantization/qnn_lower_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/quantization/qnn_lower_example.py -------------------------------------------------------------------------------- /quantization/tanh_runge_katta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/quantization/tanh_runge_katta.py -------------------------------------------------------------------------------- /relay/1x1_conv_to_gemm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/1x1_conv_to_gemm.py -------------------------------------------------------------------------------- /relay/debug_adam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/debug_adam.py -------------------------------------------------------------------------------- /relay/debug_conv2d_transpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/debug_conv2d_transpose.py -------------------------------------------------------------------------------- /relay/debug_qnn_grouped_convs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/debug_qnn_grouped_convs.py -------------------------------------------------------------------------------- /relay/depthwise_convolution_inconsistency.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/depthwise_convolution_inconsistency.py -------------------------------------------------------------------------------- /relay/dynamic_to_static_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/dynamic_to_static_example.py -------------------------------------------------------------------------------- /relay/example_fold_constant_pass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/example_fold_constant_pass.py -------------------------------------------------------------------------------- /relay/example_list_all_ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/example_list_all_ops.py -------------------------------------------------------------------------------- /relay/graph_debugger_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/graph_debugger_example.py -------------------------------------------------------------------------------- /relay/import_onnx_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/import_onnx_model.py -------------------------------------------------------------------------------- /relay/inputs.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/inputs.pkl -------------------------------------------------------------------------------- /relay/inputs2.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/inputs2.pkl -------------------------------------------------------------------------------- /relay/inputs3.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/inputs3.pkl -------------------------------------------------------------------------------- /relay/nhwc_layout_transform_example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/nhwc_layout_transform_example.py -------------------------------------------------------------------------------- /relay/test_batch_matmul.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/test_batch_matmul.py -------------------------------------------------------------------------------- /relay/test_dropout.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/test_dropout.py -------------------------------------------------------------------------------- /relay/test_layout_transform_no_effect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/test_layout_transform_no_effect.py -------------------------------------------------------------------------------- /relay/test_layout_transform_pass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/test_layout_transform_pass.py -------------------------------------------------------------------------------- /relay/test_multiply_by_0_nan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/test_multiply_by_0_nan.py -------------------------------------------------------------------------------- /relay/test_multiply_by_0_nan_data/inputs.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/test_multiply_by_0_nan_data/inputs.pkl -------------------------------------------------------------------------------- /relay/test_multiply_by_0_nan_data/inputs2.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/test_multiply_by_0_nan_data/inputs2.pkl -------------------------------------------------------------------------------- /relay/test_multiply_by_0_nan_data/inputs3.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/test_multiply_by_0_nan_data/inputs3.pkl -------------------------------------------------------------------------------- /relay/test_reshape_extraneous.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/relay/test_reshape_extraneous.py -------------------------------------------------------------------------------- /resnet-18-NCHW-B1-llvm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/resnet-18-NCHW-B1-llvm.json -------------------------------------------------------------------------------- /resnet-18-NHWC-B1-llvm.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/resnet-18-NHWC-B1-llvm.json -------------------------------------------------------------------------------- /runtime/test_benchmark_funcs_graph_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/runtime/test_benchmark_funcs_graph_runtime.py -------------------------------------------------------------------------------- /tir/lowering_to_tir_autoscheduler_from_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/tir/lowering_to_tir_autoscheduler_from_logs.py -------------------------------------------------------------------------------- /tir/lowering_to_tir_autotvm_from_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/tir/lowering_to_tir_autotvm_from_logs.py -------------------------------------------------------------------------------- /tir/te_matrix_multiplication.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/tir/te_matrix_multiplication.py -------------------------------------------------------------------------------- /tir/test_inconsistent_tir_lowering.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/tir/test_inconsistent_tir_lowering.py -------------------------------------------------------------------------------- /tir/test_inconsistent_tir_printing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/tir/test_inconsistent_tir_printing.py -------------------------------------------------------------------------------- /tir/test_serialize_dag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/tir/test_serialize_dag.py -------------------------------------------------------------------------------- /topi/conv2d_cuda_exploration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/AndrewZhaoLuo/TVM-Sandbox/HEAD/topi/conv2d_cuda_exploration.py --------------------------------------------------------------------------------