chore: import upstream snapshot with attribution

python/tvm/driver/__init__.py

@@ -0,0 +1,22 @@
+# isort: skip_file
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+# pylint: disable=redefined-builtin
+
+"""Namespace for driver APIs"""
+
+from .build_module import build, compile

python/tvm/driver/_ffi_api.py

@@ -0,0 +1,21 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+"""FFI APIs for tvm.driver"""
+
+import tvm_ffi
+
+tvm_ffi.init_ffi_api("driver", __name__)

python/tvm/driver/build_module.py

@@ -0,0 +1,112 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+# pylint: disable=invalid-name
+"""The build utils in python."""
+
+import warnings
+from collections.abc import Callable
+
+import tvm
+from tvm.ir.module import IRModule
+from tvm.runtime import Executable
+from tvm.target import Target
+from tvm.tirx import PrimFunc
+
+
+def build(
+    mod: PrimFunc | IRModule,
+    target: str | Target | None = None,
+    pipeline: str | tvm.transform.Pass | None = "default",
+):
+    """
+    Build a function with a signature, generating code for devices
+    coupled with target information.
+
+    This function is deprecated. Use `tvm.compile` or `tvm.tirx.build` instead.
+
+    Parameters
+    ----------
+    mod : Union[PrimFunc, IRModule]
+        The input to be built.
+    target : Optional[Union[str, Target]]
+        The target for compilation.
+    pipeline : Optional[Union[str, tvm.transform.Pass]]
+        The pipeline to use for compilation.
+
+    Returns
+    -------
+    tvm.runtime.Module
+        A module combining both host and device code.
+    """
+    warnings.warn(
+        "build is deprecated. Use `tvm.compile` or `tvm.tirx.build` instead.",
+        DeprecationWarning,
+    )
+    return tvm.tirx.build(mod, target, pipeline)
+
+
+def _contains_relax(mod: PrimFunc | IRModule) -> bool:
+    if isinstance(mod, PrimFunc):
+        return False
+    if isinstance(mod, IRModule):
+        return any(isinstance(func, tvm.relax.Function) for _, func in mod.functions_items())
+
+    raise ValueError(f"Function input must be a PrimFunc or IRModule, but got {type(mod)}")
+
+
+def compile(  # pylint: disable=redefined-builtin
+    mod: PrimFunc | IRModule,
+    target: Target | None = None,
+    *,
+    relax_pipeline: tvm.transform.Pass | Callable | str | None = "default",
+    tir_pipeline: tvm.transform.Pass | Callable | str | None = "default",
+) -> Executable:
+    """
+    Compile an IRModule to a runtime executable.
+
+    This function serves as a unified entry point for compiling both TIR and Relax modules.
+    It automatically detects the module type and routes to the appropriate build function.
+
+    Parameters
+    ----------
+    mod : Union[PrimFunc, IRModule]
+        The input module to be compiled. Can be a PrimFunc or an IRModule containing
+        TIR or Relax functions.
+    target : Optional[Target]
+        The target platform to compile for.
+    relax_pipeline : Optional[Union[tvm.transform.Pass, Callable, str]]
+        The compilation pipeline to use for Relax functions.
+        Only used if the module contains Relax functions.
+    tir_pipeline : Optional[Union[tvm.transform.Pass, Callable, str]]
+        The compilation pipeline to use for TIR functions.
+
+    Returns
+    -------
+    Executable
+        A runtime executable that can be loaded and executed.
+    """
+    # TODO(tvm-team): combine two path into unified one
+    if _contains_relax(mod):
+        return tvm.relax.build(
+            mod,
+            target,
+            relax_pipeline=relax_pipeline,
+            tir_pipeline=tir_pipeline,
+        )
+    lib = tvm.tirx.build(mod, target, pipeline=tir_pipeline)
+    return Executable(lib)