Matthew Barrett f7f2cda675 [AOT] Add AOTLowerMain pass to lower a Relay main into TIR (#12550) ...

This is a pass refactored out of the AOTExecutorCodegen. Instead of
combining all of the functionality of the AOTExecutorCodegen into a
single monolithic pass, this pass only handles the lowering of the
Relay main function into TIR. Tests for the pass are included.

2022-09-14 10:25:45 +01:00

..

aot

[AOT] Add AOTLowerMain pass to lower a Relay main into TIR (#12550)

2022-09-14 10:25:45 +01:00

backend

Unify name mangling in TVM (#12066)

2022-08-11 09:18:46 -07:00

benchmarking

[Refactor] Rename asnumpy -> numpy in NDArray (#8083)

2021-05-21 08:32:27 -04:00

collage

[Collage] PruneCandidates and demo_collage_partition.py (#12105)

2022-07-15 11:04:33 -07:00

dyn

[FIX] Fix bug in resize2d unittest func name (#12498)

2022-08-22 10:07:38 -07:00

op

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

opencl_texture

[OpenCL] Enable OpenCL for GPU tests (#12490)

2022-09-10 04:01:53 +09:00

qnn

[QNN] Lookup operations for hard to implement operators (#10053)

2022-02-09 10:54:54 -08:00

strategy/arm_cpu

Add Arm DSP implementation of Depthwise Conv2D (#12448)

2022-09-07 22:16:46 -07:00

transform

[BYOC] Handle constants in IRModule-at-a-time external codegen (#11770)

2022-06-30 10:27:21 -07:00

utils

[OpenCL] Enable OpenCL for GPU tests (#12490)

2022-09-10 04:01:53 +09:00

test_adt.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_analysis_basic_block_normal_form.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_analysis_extract_fake_quantized_ops.py

[Relay][Pass] Add a relay pass to extract fake quantized ops (#10089)

2022-02-01 20:57:45 -08:00

test_analysis_extract_fused_functions.py

[Relay, TOPI] Make Softmax op fusible with elemwise ops (#8909)

2021-09-06 15:30:36 +09:00

test_analysis_extract_intermediate_expr.py

[Relay] Extract intermediate node by its expression ID (#12646)

2022-09-01 19:32:42 +09:00

test_analysis_extract_operators.py

[Relay][Pass] Add ExtractOperators pass (#8996)

2021-09-16 09:49:08 +09:00

test_analysis_feature.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_analysis_get_calibration_data.py

[Refactor] Rename asnumpy -> numpy in NDArray (#8083)

2021-05-21 08:32:27 -04:00

test_annotated_regions.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_any.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_auto_scheduler_layout_rewrite_networks.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_auto_scheduler_task_extraction.py

[Autoscheduler] Configurable workload keys (#8862)

2021-08-28 02:24:16 -07:00

test_auto_scheduler_tuning.py

[Relay] Bug fix when applying history using an iterator or records. (#11306)

2022-05-17 16:17:48 -07:00

test_autotvm_task_extraction.py

[Topi][Unittests] Parametrized tests in test_topi_dense.py, split out gpu-independent implementations (#8336)

2021-06-30 13:59:16 +08:00

test_backend_graph_executor.py

[Texture] Add 2d memory support into static memory planner (#11876)

2022-07-19 05:39:59 +09:00

test_backend_interpreter.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_build_module.py

[Target] Remove deprecated parameters from target (#12416)

2022-08-24 13:02:27 +09:00

test_call_graph.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_change_batch.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_cmp_op.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_const.py

[Refactor] Rename asnumpy -> numpy in NDArray (#8083)

2021-05-21 08:32:27 -04:00

test_cpp_build_module.py

[TIR] [bfloat16] add bfloat16 promotion for CallNode (#12370)

2022-08-18 10:44:07 -07:00

test_dataflow_pattern.py

[Relay] Restore dominator check (#11616)

2022-06-08 11:00:35 +01:00

test_debug.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_executor.py

[relay] Changed 'name' field to 'registry_name' for Executor and Runtime (#10466)

2022-07-11 14:37:20 -07:00

test_expr_functor.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_external_codegen.py

[BYOC] Handle constants in IRModule-at-a-time external codegen (#11770)

2022-06-30 10:27:21 -07:00

test_ir_bind.py

[Fixbug] Report duplicated param names of relay function when bind params (#9350)

2021-10-24 15:28:31 -07:00

test_ir_module.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_ir_nodes.py

Implementation of Common Subexpression Elimination for TIR (#9482)

2022-02-10 21:46:23 +09:00

test_ir_op.py

Complete register op from python (#8079)

2021-06-04 20:48:52 +08:00

test_ir_parser.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_ir_structural_equal_hash.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_ir_text_printer.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_ir_well_formed.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_json_compact.py

[BUGFIX] Check that virtual device is unchanged in WithFields (#9826)

2022-01-12 11:35:25 +09:00

test_json_runtime.py

[CORE][Relay] Swap and remove compile_engine with te_compiler followup of #8775 (#9282)

2021-10-23 06:16:05 +09:00

test_layer_count.py

[Relay] A set of utilities that allows a model to be run efficiently on tensorcores. (#6748)

2020-10-26 19:37:05 -04:00

test_memory_passes.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_name_mangling.py

[AOT] Name mangling in AOT (#8014)

2021-06-28 08:34:49 -07:00

test_name_supply.py

Unify name mangling in TVM (#12066)

2022-08-11 09:18:46 -07:00

test_name_transforms.py

Migrate C Interface API Generation to C++ (#9106)

2021-10-08 07:40:07 +01:00

test_op_fast_math.py

[AOT] Name mangling in AOT (#8014)

2021-06-28 08:34:49 -07:00

test_op_grad_level1.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_op_grad_level2.py

[Tests] Replace the Relay interpreter with the VM in the op tests (#11386)

2022-05-23 19:14:00 +09:00

test_op_grad_level3.py

[Tests] Replace the Relay interpreter with the VM in the op tests (#11386)

2022-05-23 19:14:00 +09:00

test_op_grad_level4.py

[Tests] Replace the Relay interpreter with the VM in the op tests (#11386)

2022-05-23 19:14:00 +09:00

test_op_grad_level10.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_op_level1.py

support any shape and axis for log softmax (#11951)

2022-06-30 09:43:48 +09:00

test_op_level2.py

[DNNL] Add bfloat16 type support for dnnl conv2d kernel (#11902)

2022-06-27 20:16:20 -07:00

test_op_level3.py

[CUDA][CodeGen] Fix cuda codegen's fp16 inf literal (#12581)

2022-08-25 10:42:57 -07:00

test_op_level4.py

[TIR][Arith] Add more strict checking in imm construction and folding. (#12515)

2022-09-09 12:10:50 -07:00

test_op_level5.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_op_level6.py

[Tests] Replace the Relay interpreter with the VM in the op tests (#11386)

2022-05-23 19:14:00 +09:00

test_op_level10.py

Several type mismatch fixes and checks (#12041)

2022-07-11 10:05:21 +09:00

test_op_qnn_add.py

[QNN] Fix per-channel broadcast with invalid axes (#10936)

2022-04-08 14:01:58 +09:00

test_op_qnn_batch_matmul.py

initial commit (#12301)

2022-08-04 14:05:01 -06:00

test_op_qnn_concatenate.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_op_qnn_conv2_transpose.py

[QNN] Support input scale and zp of 1-element vector in qnn.conv2d_transpose (#10952)

2022-04-20 09:47:37 -07:00

test_op_qnn_conv2d.py

[Relay][QNN] Support for non scalar zero points in qnn.conv2d (#8620)

2021-08-07 11:26:11 +09:00

test_op_qnn_dense.py

[Refactor] Rename asnumpy -> numpy in NDArray (#8083)

2021-05-21 08:32:27 -04:00

test_op_qnn_dequantize.py

[QNN] Fix qnn.dequantize scale and zp shape (#10880)

2022-04-05 15:18:34 -07:00

test_op_qnn_leaky_relu.py

[QNN] Support different qnn params between in/out tensor in leaky_relu (#12116)

2022-07-22 14:33:31 -07:00

test_op_qnn_mul.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_op_qnn_quantize.py

[Relay][Quantization] Extend FakeQuantizationToInteger to more ops (#8241)

2021-08-03 14:53:47 +09:00

test_op_qnn_requantize.py

Add FP requantize flow. Set float32 flow by default for llvm x86 targets with (#9637)

2022-01-26 21:22:33 +09:00

test_op_qnn_simulated_dequantize.py

[VM] Bug fix for numpy scalar input in vm (#8553)

2021-07-30 14:11:51 +09:00

test_op_qnn_simulated_quantize.py

[VM] Bug fix for numpy scalar input in vm (#8553)

2021-07-30 14:11:51 +09:00

test_op_qnn_subtract.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_op_qnn_unary_elementwise.py

[QNN] Add hardswish int8 impl using table lookup (#11700)

2022-06-28 10:10:51 -07:00

test_param_dict.py

[Refactor] Rename asnumpy -> numpy in NDArray (#8083)

2021-05-21 08:32:27 -04:00

test_pass_alter_op_layout.py

Fix infercorrect layout in Layoutrewrite and improve naming. (#12007)

2022-07-06 10:50:25 +09:00

test_pass_annotate_spans_defuse.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_pass_annotate_target.py

Use std::string_view, remove experimental or pre-14 variants, NFC (#12460)

2022-08-16 21:38:29 -07:00

test_pass_auto_quantize.py

[FIX] Fix bug in MobileNetV2 quantization (#8243)

2022-01-10 07:35:59 +09:00

test_pass_canonicalize_cast.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_pass_check_kind.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_pass_collage_partition.py

[Collage] PruneCandidates and demo_collage_partition.py (#12105)

2022-07-15 11:04:33 -07:00

test_pass_combine_parallel_batch_matmul.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_pass_combine_parallel_conv2d.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_pass_combine_parallel_dense.py

[AutoScheduler] Add layout rewrite support for dense and batch matmul on CPU (#7161)

2020-12-24 16:57:24 -08:00

test_pass_convert_op_layout.py

[Relay][Layout] Add FInferCorrectLayout for L2 norm layout transform. (#12497)

2022-08-21 04:54:22 +09:00

test_pass_dead_code_elimination.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_pass_defunctionalization.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_pass_defuse_ops.py

[Relay][DefuseOps pass] bug fix: To support function body types other than call node (#10069)

2022-01-26 21:25:15 +09:00

test_pass_dynamic_to_static.py

[ONNX][Relay] Support "tf_crop_and_resize" in relay Resize op. (#9475)

2021-11-16 14:04:56 +09:00

test_pass_eliminate_common_subexpr.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_pass_eta_expand.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_pass_fake_quantization_to_integer.py

[FQ2I] Quantized constant bias (#12666)

2022-09-13 17:24:33 -07:00

test_pass_fast_math.py

[FastMath] Add fast_softmax support in fast_math pass (#8138)

2021-05-26 09:29:15 -07:00

test_pass_flatten_atrous_conv.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_pass_flexible_shape_dispatch.py

[Relay] Flexible shape dispatch transformation (#11199)

2022-05-06 12:18:16 -07:00

test_pass_fold_constant.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_pass_fold_explicit_padding.py

fix uint case (#11597)

2022-06-07 16:58:14 +09:00

test_pass_fold_scale_axis.py

[BYOC-DNNL] enable conv3d->bn folding (#10837)

2022-04-11 11:49:13 +09:00

test_pass_fuse_ops.py

[TIR][Arith] Add more strict checking in imm construction and folding. (#12515)

2022-09-09 12:10:50 -07:00

test_pass_gradient.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_pass_inline.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_pass_instrument.py

[Bugfix] fix the bug that occurs when the test_pass_ctx_exception() is (#9774)

2021-12-22 13:22:07 +09:00

test_pass_lambda_lift.py

[ONNX] NMS in ONNX (#6839)

2020-12-14 10:54:07 -08:00

test_pass_lazy_gradient_init.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_pass_legalize_tensorcore.py

[ONNX] Add MatMulInteger importer (#10450)

2022-04-13 13:25:26 +09:00

test_pass_legalize.py

fix py files (#8194)

2021-06-04 23:22:46 +01:00

test_pass_mac_count.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_pass_manager.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_pass_manifest_lifetimes.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_pass_merge_compiler_regions.py

[REFACTOR] Remainings of util => utils (#6778)

2020-10-29 13:46:31 -04:00

test_pass_merge_composite.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_pass_partial_eval.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_pass_partition_graph.py

Use std::string_view, remove experimental or pre-14 variants, NFC (#12460)

2022-08-16 21:38:29 -07:00

test_pass_plan_devices.py

[Relay] Allow partial virtual device annotations. (#12107)

2022-07-15 14:34:28 -07:00

test_pass_qnn_legalize.py

Rename GraphRuntime to GraphExecutor (#7653)

2021-03-29 20:07:21 -04:00

test_pass_remove_unused_functions.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_pass_simplify_expr.py

[SimplifyExpr] Add simplify for dq->arg funcs (#12580)

2022-08-26 09:28:20 -07:00

test_pass_simplify_inference.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_pass_split_args.py

[Metal] Add pass for splitting kernel with huge number of args (#8313)

2021-07-01 12:14:17 +09:00

test_pass_to_a_normal_form.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_pass_to_basic_block_normal_form.py

[BUG] ToBasicBlockNormalForm immutability (#8778)

2021-08-25 15:34:10 -07:00

test_pass_to_cps.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_pass_to_graph_normal_form.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_pass_unmatched_cases.py

[Bugfix][Relay] Crash in match_exhaustion.cc when given an empty tuple pattern or constructor with no args (#7459)

2021-02-16 22:24:56 -08:00

test_pass_vars.py

[Diagnostics][Relay][InferType] Refactor InferType to work on whole module, and use new diagnostics. (#6274)

2020-10-09 11:44:09 -07:00

test_pipeline_executor.py

[Runtime][PipelineExecutor] Added Interface to Track Number of Global Inputs (#11315)

2022-06-18 04:36:31 +09:00

test_prng.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_py_converter.py

[Refactor] Rename asnumpy -> numpy in NDArray (#8083)

2021-05-21 08:32:27 -04:00

test_recast.py

Fix recast of relay ops without attributes (#8043)

2021-05-17 09:05:32 -07:00

test_relay_te_compiler.py

[tests] add utility to replace direct call to pytest.main (#11393)

2022-05-23 11:52:22 -07:00

test_runtime.py

[TVMC][Relay] Introduce executor and runtime parameters (#9352)

2021-11-19 17:11:40 -08:00

test_simplify_fc_transpose.py

[Refactor] Rename asnumpy -> numpy in NDArray (#8083)

2021-05-21 08:32:27 -04:00

test_sparse_conv2d_convert.py

[TOPI] [Relay] Sparse Conv2d Implementation for 3x3 kernels (#8605)

2021-08-27 09:18:34 +08:00

test_sparse_dense_convert.py

[Refactor] Rename asnumpy -> numpy in NDArray (#8083)

2021-05-21 08:32:27 -04:00

test_target_hooks.py

[Relay] Plumb external codegen target via Target.current() (#11432)

2022-06-01 11:51:35 -07:00

test_tensor_array.py

[Relay] Refactor Interpreter to treat lowering as IRModule->IRModule rewrite. (#8597)

2021-08-17 16:41:42 -07:00

test_to_mixed_precision.py

Fix mixed precision output type to original type (#11142)

2022-05-05 23:34:30 +08:00

test_type_functor.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_type_infer.py

[BugFix] Use shape dtype on ArgReduce to determine return type (#12083)

2022-07-14 10:54:56 -07:00

test_type_solver.py

[BugFix] resolve integer 32. ~ 64. mismatch by casting (#9582)

2022-01-06 11:12:41 -05:00

test_typecall.py

[Format] Convert all Python code w/o CI (#6448)

2020-09-11 22:17:24 +09:00

test_used_memory_annotator.py

[AOT] Calculate used memory at the callsite of primitive functions (#11208)

2022-06-25 13:03:47 +01:00

test_vm_serialization.py

[Relay] Use LowerTEPass in VM (#9483)

2021-11-30 23:15:07 -08:00

test_vm.py

Add methods to get and set late-bound constants. (#12664)

2022-09-01 08:57:40 -07:00