/usr/lib/python3.12/site-packages/torch/cuda/__init__.py:68: FutureWarning: The pynvml package is deprecated. Please install nvidia-ml-py instead. If you did not install pynvml directly, please report this to the maintainers of the package that installed pynvml for you.
  import pynvml  # type: ignore[import]
usage: tvm-ffi-stubgen [-h] [--imports IMPORTS] [--dlls LIBS] [--init-pypkg INIT_PYPKG]
                       [--init-lib INIT_LIB] [--init-prefix INIT_PREFIX] [--indent INDENT]
                       [--verbose] [--dry-run]
                       [PATH ...]

Generate type stubs for TVM FFI extensions. It supports two modes
- In `--init-*` mode, it generates missing `_ffi_api.py` and `__init__.py` files, based on the registered global functions and object types in the loaded libraries.
- In normal mode, it processes the given files/directories in-place, generating type stubs inside special `tvm-ffi-stubgen` directive blocks.

Documentation: https://tvm.apache.org/ffi/packaging/stubgen.html.

positional arguments:
  PATH                  Files or directories to process. Directories are scanned recursively; only .py and .pyi files are modified. Use tvm-ffi-stubgen directives to select where stubs are generated. (default: None)

options:
  -h, --help            show this help message and exit
  --imports IMPORTS     Additional imports to load before generation, separated by ';' (e.g. 'pkgA;pkgB.submodule'). (default: )
  --dlls LIBS           Shared libraries to preload before generation (e.g. TVM runtime or your extension), separated by ';'. This ensures global function and object metadata is available. Platform-specific suffixes like .so/.dylib/.dll are supported. (default: )
  --init-pypkg INIT_PYPKG
                        Python package name to generate stubs for (e.g. apache-tvm-ffi). Required together with --init-lib and --init-prefix. (default: )
  --init-lib INIT_LIB   CMake target that produces the shared library to load for stub generation (e.g. tvm_ffi_shared). Required together with --init-pypkg and --init-prefix. (default: )
  --init-prefix INIT_PREFIX
                        Global function/object prefix to include when generating stubs (e.g. tvm_ffi.). Required together with --init-pypkg and --init-lib. (default: )
  --indent INDENT       Extra spaces added inside each generated block, relative to the indentation of the corresponding '# tvm-ffi-stubgen(begin):' line. (default: 4)
  --verbose             Print a unified diff of changes to each file. This is useful for debugging or previewing changes before applying them. (default: False)
  --dry-run             Don't write changes to files. This is useful for previewing changes without modifying any files. (default: False)