This is an automated email from the ASF dual-hosted git repository. tqchen pushed a commit to branch tqchen/printer-refinement in repository https://gitbox.apache.org/repos/asf/tvm.git
commit f7ae6a96174404838257ea2703f4fffcfbddd442 Author: Tianqi Chen <[email protected]> AuthorDate: Thu Oct 1 16:54:53 2026 +0000 [TEST][Script] Expect fully qualified operator printer names --- tests/python/tirx/codegen/test_ptx_addr.py | 2 +- tests/python/tirx/codegen/test_ptx_dialect.py | 2 +- tests/python/tirx/test_op_namespace_cleanup.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/python/tirx/codegen/test_ptx_addr.py b/tests/python/tirx/codegen/test_ptx_addr.py index 77e3f83e89..a3b8449c4f 100644 --- a/tests/python/tirx/codegen/test_ptx_addr.py +++ b/tests/python/tirx/codegen/test_ptx_addr.py @@ -51,7 +51,7 @@ def test_ptx_addr_registration_and_table_capabilities(): op = Op.get("tirx.ptx.addr") assert int(op.get_attr("TCallEffectKind")) == CallEffectKind.Pure.value - assert op.get_attr("TScriptPrinterName") == "ptx.addr" + assert op.get_attr("TScriptPrinterName") == "tirx.ptx.addr" assert "tirx.ptx.addr" in CODEGEN_REGISTRY addresses = [slot for entry in TABLE.values() for slot in entry.operands if slot.kind == "addr"] diff --git a/tests/python/tirx/codegen/test_ptx_dialect.py b/tests/python/tirx/codegen/test_ptx_dialect.py index 7868a9edfd..20c2638204 100644 --- a/tests/python/tirx/codegen/test_ptx_dialect.py +++ b/tests/python/tirx/codegen/test_ptx_dialect.py @@ -67,7 +67,7 @@ def test_ptx_registration(): # *escaped* family: `and`/`or`/`not` (ISA 9.7.9) are Python keywords and # print as `T.ptx.and_`. Identity for every other family. family = escape_token(entry.family) # several entries may share a mnemonic - assert op.get_attr("TScriptPrinterName") == f"ptx.{family}", entry.name + assert op.get_attr("TScriptPrinterName") == f"tirx.ptx.{family}", entry.name assert entry.op_name in CODEGEN_REGISTRY, entry.name diff --git a/tests/python/tirx/test_op_namespace_cleanup.py b/tests/python/tirx/test_op_namespace_cleanup.py index 08d4f4e09f..8742546b81 100644 --- a/tests/python/tirx/test_op_namespace_cleanup.py +++ b/tests/python/tirx/test_op_namespace_cleanup.py @@ -393,8 +393,8 @@ def test_registered_tirx_ops_have_exactly_one_category(): assert device_namespace in device_namespaces, op_name printer_name = _op_attr(op_name, "TScriptPrinterName") assert printer_name is not None, op_name - assert printer_name.startswith(device_namespace + "."), op_name - assert _has_path(T, printer_name), op_name + assert printer_name.startswith("tirx." + device_namespace + "."), op_name + assert _has_path(T, printer_name.removeprefix("tirx.")), op_name else: assert category == "builtin" assert device_namespace is None, op_name
