This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a change to branch script/canonical-parser-df
in repository https://gitbox.apache.org/repos/asf/tvm.git
from e62f985078 Declare VNNI tensorization reference storage explicitly
new 5e56b88a72 [DOCS][Script] Restore builder API details and executable
JIT examples
new 8331cf160d [REFACTOR][TIRx] Consolidate shared builder protocol imports
new 36d5fe1903 [REFACTOR][Script] Type parser contracts and unify argument
policies
The 3 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails. The revisions
listed as "add" were already present in the repository and have only
been added to this reference.
Summary of changes:
.github/workflows/main.yml | 4 +-
python/tvm/relax/script/builder/__init__.py | 50 ++-
python/tvm/relax/script/builder/ir.py | 72 +++-
python/tvm/script/ir_builder/base.py | 24 +-
python/tvm/script/ir_builder/ir/ir.py | 37 +-
python/tvm/script/parser/__init__.py | 16 +-
python/tvm/script/parser/call_args_policy.py | 29 +-
python/tvm/script/parser/diagnostics.py | 6 +-
python/tvm/script/parser/entry.py | 228 +++++++++---
python/tvm/script/parser/jit_support.py | 27 +-
python/tvm/script/parser/prescan.py | 108 +++---
python/tvm/script/parser/protocol.py | 228 ++++--------
python/tvm/script/parser/source.py | 14 +-
python/tvm/script/parser/transpile.py | 234 +++++++-----
python/tvm/tirx/script/builder/__init__.py | 58 ++-
python/tvm/tirx/script/builder/ir.py | 197 ++++++++--
python/tvm/tirx/script/builder/protocol.py | 44 +--
python/tvm/tirx/script/builder/tirx.py | 32 +-
python/tvm/tirx/script/jit.py | 39 ++
.../test_capture_annotation_constructors.py} | 0
.../script/test_capture_annotation_identity.py | 87 +++++
.../test_frontend_relax_construction_validation.py | 77 ++++
.../script/test_parser_captured_shapes.py} | 30 +-
.../script/test_parser_construction_regressions.py | 68 ++++
.../test_parser_distributed_aliases.py | 0
.../script/test_parser_distributed_range.py} | 40 +-
.../test_parser_local_function_types.py | 0
.../{ => script}/test_parser_local_functions.py | 0
.../script/test_parser_relax_emission.py} | 44 +--
.../script/test_parser_relax_frames.py} | 170 ++-------
.../script/test_parser_relax_resolvers.py} | 69 +---
.../test_relax_parser_comparison_integration.py | 90 +++++
.../test_frontend_nn_llm_script_helpers.py} | 98 +----
tests/python/relax/test_parser_scoped_symbols.py | 96 -----
.../test_s_tir_axis_binding_validation.py} | 41 +--
.../test_s_tir_migrated_source_symbols.py} | 2 +-
.../tile_primitive/cuda => script}/conftest.py | 29 +-
tests/{parser => python/script}/dummy_builder.py | 19 +-
.../python/script/test_capture_definition_scope.py | 402 +++++++++++++++++++++
tests/python/script/test_capture_frontend_scope.py | 232 ++++++++++++
.../test_capture_gallery_source.py} | 74 ++--
tests/python/script/test_capture_nonlocal.py | 66 ++++
tests/{parser => python/script}/test_parser.py | 33 +-
.../test_parser_bypass_source_attachment.py} | 37 +-
.../{tirx => script}/test_parser_call_scopes.py | 86 +++--
.../script/test_parser_comparison_semantics.py | 347 ++++++++++++++++++
tests/python/script/test_parser_host_selection.py | 263 ++++++++++++++
tests/python/script/test_parser_scoped_symbols.py | 104 ++++++
.../python/script/test_parser_shared_resolvers.py | 79 ++++
tests/python/script/test_parser_span_helpers.py | 72 ++++
.../script/test_capture_jit_execution.py} | 53 ++-
.../test_frontend_construction_validation.py | 96 +++++
.../{ => script}/test_parser_backend_namespaces.py | 0
.../script/test_parser_buffer_strides.py} | 81 +----
.../tirx/{ => script}/test_parser_buffer_views.py | 0
.../tirx/script/test_parser_constexpr_buffers.py | 50 +++
.../tirx/{ => script}/test_parser_emitted_spans.py | 42 +--
.../{ => script}/test_parser_helper_migrations.py | 83 +----
.../{ => script}/test_parser_jit_specialization.py | 0
.../{ => script}/test_parser_printer_type_vars.py | 0
.../{ => script}/test_parser_range_annotations.py | 11 +-
.../{ => script}/test_parser_scope_id_identity.py | 17 +-
.../{ => script}/test_parser_scope_id_names.py | 0
.../tirx/{ => script}/test_parser_self_bind.py | 4 +-
.../{ => script}/test_parser_tile_namespaces.py | 0
.../script/test_parser_tirx_emission.py} | 23 +-
.../script/test_parser_tirx_frames.py} | 187 +++-------
.../script/test_parser_tirx_resolvers.py} | 76 +---
.../test_tirx_parser_comparison_integration.py | 199 ++++++++++
tests/python/tirx/test_parser_constexpr.py | 264 --------------
.../test_parser_comparison_source_order.py | 298 ---------------
.../unittest/test_parser_comparison_values.py | 203 -----------
.../unittest/test_parser_definition_context.py | 391 --------------------
.../unittest/test_parser_frontend_regressions.py | 303 ----------------
tests/python/unittest/test_parser_nonlocal.py | 57 ---
tests/scripts/task_python_unittest.sh | 2 +-
76 files changed, 3531 insertions(+), 3111 deletions(-)
rename tests/python/{unittest/test_parser_captured_constructors.py =>
relax/script/test_capture_annotation_constructors.py} (100%)
create mode 100644
tests/python/relax/script/test_capture_annotation_identity.py
create mode 100644
tests/python/relax/script/test_frontend_relax_construction_validation.py
copy tests/python/{ir/test_roundtrip_runtime_module.py =>
relax/script/test_parser_captured_shapes.py} (62%)
create mode 100644
tests/python/relax/script/test_parser_construction_regressions.py
rename tests/python/relax/{ => script}/test_parser_distributed_aliases.py
(100%)
rename tests/python/{unittest/test_parser_ir_namespace.py =>
relax/script/test_parser_distributed_range.py} (58%)
rename tests/python/relax/{ => script}/test_parser_local_function_types.py
(100%)
rename tests/python/relax/{ => script}/test_parser_local_functions.py (100%)
copy tests/python/{unittest/test_parser_emission_contract.py =>
relax/script/test_parser_relax_emission.py} (62%)
copy tests/python/{unittest/test_parser_native_frames.py =>
relax/script/test_parser_relax_frames.py} (59%)
copy tests/python/{unittest/test_parser_support.py =>
relax/script/test_parser_relax_resolvers.py} (55%)
create mode 100644
tests/python/relax/script/test_relax_parser_comparison_integration.py
copy tests/python/{tirx/test_parser_helper_migrations.py =>
relax/test_frontend_nn_llm_script_helpers.py} (51%)
delete mode 100644 tests/python/relax/test_parser_scoped_symbols.py
copy tests/python/{ir/test_dtype.py =>
s_tir/test_s_tir_axis_binding_validation.py} (56%)
rename tests/python/{tirx/test_parser_dtype_migrations.py =>
s_tir/test_s_tir_migrated_source_symbols.py} (98%)
copy tests/python/{tirx/operator/tile_primitive/cuda => script}/conftest.py
(59%)
rename tests/{parser => python/script}/dummy_builder.py (93%)
create mode 100644 tests/python/script/test_capture_definition_scope.py
create mode 100644 tests/python/script/test_capture_frontend_scope.py
rename tests/python/{unittest/test_parser_class_source.py =>
script/test_capture_gallery_source.py} (55%)
create mode 100644 tests/python/script/test_capture_nonlocal.py
rename tests/{parser => python/script}/test_parser.py (97%)
copy tests/python/{tvmscript/test_tvmscript_ir_builder_irmodule.py =>
script/test_parser_bypass_source_attachment.py} (60%)
rename tests/python/{tirx => script}/test_parser_call_scopes.py (58%)
create mode 100644 tests/python/script/test_parser_comparison_semantics.py
create mode 100644 tests/python/script/test_parser_host_selection.py
create mode 100644 tests/python/script/test_parser_scoped_symbols.py
create mode 100644 tests/python/script/test_parser_shared_resolvers.py
create mode 100644 tests/python/script/test_parser_span_helpers.py
copy tests/python/{ir/test_datatype_nv_fp4.py =>
tirx/script/test_capture_jit_execution.py} (50%)
create mode 100644
tests/python/tirx/script/test_frontend_construction_validation.py
rename tests/python/tirx/{ => script}/test_parser_backend_namespaces.py (100%)
rename tests/python/{relax/test_parser_construction_regressions.py =>
tirx/script/test_parser_buffer_strides.py} (58%)
rename tests/python/tirx/{ => script}/test_parser_buffer_views.py (100%)
create mode 100644 tests/python/tirx/script/test_parser_constexpr_buffers.py
rename tests/python/tirx/{ => script}/test_parser_emitted_spans.py (74%)
rename tests/python/tirx/{ => script}/test_parser_helper_migrations.py (60%)
rename tests/python/tirx/{ => script}/test_parser_jit_specialization.py (100%)
rename tests/python/tirx/{ => script}/test_parser_printer_type_vars.py (100%)
rename tests/python/tirx/{ => script}/test_parser_range_annotations.py (93%)
rename tests/python/tirx/{ => script}/test_parser_scope_id_identity.py (93%)
rename tests/python/tirx/{ => script}/test_parser_scope_id_names.py (100%)
rename tests/python/tirx/{ => script}/test_parser_self_bind.py (97%)
rename tests/python/tirx/{ => script}/test_parser_tile_namespaces.py (100%)
rename tests/python/{unittest/test_parser_emission_contract.py =>
tirx/script/test_parser_tirx_emission.py} (83%)
rename tests/python/{unittest/test_parser_native_frames.py =>
tirx/script/test_parser_tirx_frames.py} (70%)
rename tests/python/{unittest/test_parser_support.py =>
tirx/script/test_parser_tirx_resolvers.py} (52%)
create mode 100644
tests/python/tirx/script/test_tirx_parser_comparison_integration.py
delete mode 100644 tests/python/tirx/test_parser_constexpr.py
delete mode 100644 tests/python/unittest/test_parser_comparison_source_order.py
delete mode 100644 tests/python/unittest/test_parser_comparison_values.py
delete mode 100644 tests/python/unittest/test_parser_definition_context.py
delete mode 100644 tests/python/unittest/test_parser_frontend_regressions.py
delete mode 100644 tests/python/unittest/test_parser_nonlocal.py