Bohan Hou b6db2ec89f [Runtime] CutensorMap support (#18097) ...

This PR introduces Cutensor map support in the runtime module. It enables calling kernels whose arguments are cuTensorMap, these arguments are passed as handle(address) and associated with arg_extra_tags that indicate indicate it is tensor map. The TensorMap is allocated on stack with a runtime API

2025-06-30 07:59:52 -04:00

..

ir_builder

[Runtime] CutensorMap support (#18097)

2025-06-30 07:59:52 -04:00

parser

[SCRIPT] Bump Python minimum version to 3.9 and update AST compatibility (#18086)

2025-06-23 13:04:07 -04:00

printer

[REFACTOR][PYTHON] Phase out tvm._ffi and Limited API support (#18020)

2025-05-28 16:52:36 -04:00

__init__.py

[Unity][TVMScript] Move tir/relax import in script out of __init__.py (#14033)

2023-04-01 15:31:37 -04:00

_ffi_api.py

[REFACTOR][PYTHON] Phase out tvm._ffi and Limited API support (#18020)

2025-05-28 16:52:36 -04:00

highlight.py

[TVMScript] Optionally use ruff format instead of black (#16876)

2024-04-17 10:34:10 +05:30

relax.py

[Unity][TVMScript] Move tir/relax import in script out of __init__.py (#14033)

2023-04-01 15:31:37 -04:00

tir.py

[Unity][TVMScript] Move tir/relax import in script out of __init__.py (#14033)

2023-04-01 15:31:37 -04:00