MCPcopy Create free account

hub / github.com/apache/tvm-ffi / functions

Functions5,333 in github.com/apache/tvm-ffi

↓ 2 callersFunction_discover_cpp_variants
Discover available C++ compiler variants (for __dso_handle tests).
addons/tvm_ffi_orcjit/tests/test_memory_manager.py:138
↓ 2 callersFunction_extra_cflags
Return extra compiler flags for the current platform.
addons/tvm_ffi_orcjit/examples/quick-start/run.py:45
↓ 2 callersFunction_field_owner_classes
Classes whose annotations become this type's own fields.
python/tvm_ffi/dataclasses/py_class.py:152
↓ 2 callersFunction_find_cuda_home
Find the CUDA install path.
python/tvm_ffi/cpp/extension.py:136
↓ 2 callersFunction_find_libdir
Find the library directory for tvm-ffi.
python/tvm_ffi/config.py:26
↓ 2 callersFunction_find_llvm_clang
Find LLVM clang/clang++ via LLVM_PREFIX or PATH. Returns (clang, clang++) paths, or None if not found.
addons/tvm_ffi_orcjit/tests/utils.py:133
↓ 2 callersFunction_find_or_insert_file
(path: Path)
python/tvm_ffi/stub/cli.py:122
↓ 2 callersFunction_flush_pending
Retry all pending classes. Called after each successful phase 2.
python/tvm_ffi/dataclasses/py_class.py:500
↓ 2 callersMethod_format_field_params
Format init-eligible fields as a parameter string with defaults and kw_only.
python/tvm_ffi/stub/utils.py:333
↓ 2 callersFunction_get_libc
Get a ctypes handle to libc with correct mmap/munmap signatures.
addons/tvm_ffi_orcjit/tests/test_memory_manager.py:200
↓ 2 callersFunction_hash_sources
Generate a unique hash for the given sources and functions.
python/tvm_ffi/cpp/extension.py:68
↓ 2 callersFunction_install_ffi_init_attr
Install ``__ffi_init__`` as a method that delegates to ``__init_handle_by_constructor__``. Custom ``__init__`` methods call ``self.__ffi_init__(*
python/tvm_ffi/registry.py:421
↓ 2 callersFunction_instantiate
Construct a subclass instance with auto-assigned ``_value``/``_name``.
python/tvm_ffi/dataclasses/enum.py:962
↓ 2 callersFunction_is_class_var
Return True if *annotation* is ``ClassVar`` or ``ClassVar[...]``.
python/tvm_ffi/dataclasses/enum.py:1054
↓ 2 callersFunction_is_lib_prebuilt
()
addons/torch_c_dlpack_ext/build_backend.py:41
↓ 2 callersMethod_make_cls
(self)
tests/python/test_dataclass_copy.py:1123
↓ 2 callersFunction_make_init
Build ``__init__`` that delegates to ``__ffi_init__``. Both ``@c_class`` and ``@py_class`` use the same constructor-call path (``self.__init_
python/tvm_ffi/_dunder.py:31
↓ 2 callersFunction_make_line
(mod: str, items: list[ImportItem], indent: int)
python/tvm_ffi/stub/codegen.py:168
↓ 2 callersFunction_make_nested_singleton_array
(depth: int)
tests/python/test_dataclass_hash.py:655
↓ 2 callersFunction_make_nested_singleton_array
(depth: int)
tests/python/test_dataclass_compare.py:1176
↓ 2 callersFunction_make_str_enum_type
()
tests/python/test_type_converter.py:78
↓ 2 callersFunction_measure_distance_under_pressure
Load two objects under VA pressure and return (distance, overflowed). Returns ``(distance_bytes, False)`` when both objects load successfully,
addons/tvm_ffi_orcjit/tests/test_memory_manager.py:321
↓ 2 callersMethod_path_equal
(self, _1: AccessPath, /)
python/tvm_ffi/access_path.py:102
↓ 2 callersFunction_prepare_payload_enum_subclass
Inject and validate the user-visible ``value`` field for payload enums.
python/tvm_ffi/dataclasses/enum.py:1077
↓ 2 callersFunction_register
Register the object type with the FFI core.
python/tvm_ffi/registry.py:78
↓ 2 callersFunction_resolve_gpu_backend
(backend: str | None)
python/tvm_ffi/cpp/extension.py:60
↓ 2 callersFunction_rollback_registration
Undo :func:`_register_type_without_fields` after a phase-2 failure. The C-level type index is permanently consumed (cannot be reclaimed), but
python/tvm_ffi/dataclasses/py_class.py:116
↓ 2 callersFunction_sequence_compare_other
Normalize plain Python sequences for structural container equality.
python/tvm_ffi/container.py:84
↓ 2 callersFunction_sequence_contains
Containment with a Python-level structural fallback for nested sequences.
python/tvm_ffi/container.py:96
↓ 2 callersFunction_split_list_arg
(arg: str | None)
python/tvm_ffi/stub/cli.py:258
↓ 2 callersFunction_stage_2
( files: list[FileInfo], ty_map: dict[str, str], init_cfg: InitConfig, init_path: Path, gl
python/tvm_ffi/stub/cli.py:115
↓ 2 callersFunction_stage_3
( # noqa: PLR0912 file: FileInfo, opt: Options, ty_map: dict[str, str], global_funcs: dict[st
python/tvm_ffi/stub/cli.py:188
↓ 2 callersFunction_type_suffix_and_record
( ty_map: dict[str, str], imports: list[ImportItem], func_names: set[str] | None = None, )
python/tvm_ffi/stub/codegen.py:28
↓ 2 callersFunction_unique_type_key
(base: str)
tests/python/test_type_converter.py:66
↓ 2 callersFunction_validate_argument_names
Validate that argument names are valid Python identifiers and unique. Parameters ---------- names List of argument names to valid
python/tvm_ffi/utils/kwargs_wrapper.py:48
↓ 2 callersMethodadd_i64
(self, _1: int, /)
python/tvm_ffi/testing/testing.py:76
↓ 2 callersFunctionadd_one
Add one to the input integer.
python/tvm_ffi/testing/testing.py:196
↓ 2 callersMethodat
! \brief Get the i-th element of the shape. */
include/tvm/ffi/container/shape.h:91
↓ 2 callersMethodat
! * \brief Immutably read i-th element from the shape tuple. * \param idx The index * \return the i-th element. */
include/tvm/ffi/container/shape.h:282
↓ 2 callersMethodbacktrace
Get the backtrace of the error # Returns The backtrace of the error
rust/tvm-ffi/src/error.rs:136
↓ 2 callersMethodbegin
include/tvm/ffi/container/map_base.h:1466
↓ 2 callersMethodbegin
! \return begin iterator */
include/tvm/ffi/container/dict.h:244
↓ 2 callersFunctionbenchmark_call
Benchmark a function by calling it multiple times. Parameters ---------- name : str Name of the benchmark func : callable
examples/cubin_launcher/benchmark_overhead.py:86
↓ 2 callersFunctionbuild_ninja
Build the module in the given build directory using ninja.
python/tvm_ffi/cpp/extension.py:586
↓ 2 callersFunctionbuild_test_objects
Build test objects for all available compiler variants. Objects are placed under *out_dir* (default: ``{tempdir}/tvm_ffi_orcjit_tests``). Al
addons/tvm_ffi_orcjit/tests/utils.py:160
↓ 2 callersMethodcall_tuple
(&self, tuple_args: TupleType)
rust/tvm-ffi/src/function.rs:126
↓ 2 callersFunctionclassifyOverflowSections
! \brief Identify sections eligible for the overflow (separate-mmap) * path. * * Name-based candidate selection followed by edge validatio
addons/tvm_ffi_orcjit/src/ffi/orcjit_slab.cc:119
↓ 2 callersMethodconflict_obj
Return path prefix for test_funcs_conflict object.
addons/tvm_ffi_orcjit/tests/test_basic.py:91
↓ 2 callersFunctionconvert
Convert a Python object into TVM FFI values. This helper mirrors the automatic argument conversion that happens when calling FFI functions. I
python/tvm_ffi/_convert.py:40
↓ 2 callersFunctioncopyright_line
(line: str)
tests/lint/check_asf_header.py:202
↓ 2 callersFunctioncreate_tensor
Helper to create a Tensor with a specific float value and shape
rust/tvm-ffi/tests/test_array.rs:22
↓ 2 callersMethodctor_dtor_obj
Return path prefix for test_ctor_dtor object.
addons/tvm_ffi_orcjit/tests/test_basic.py:115
↓ 2 callersMethoddecommitPages
addons/tvm_ffi_orcjit/src/ffi/orcjit_slab.cc:188
↓ 2 callersMethoddtype
! * \brief Get the data type of the Tensor. * \return The data type of the Tensor. */
include/tvm/ffi/container/tensor.h:305
↓ 2 callersMethodend
include/tvm/ffi/container/map_base.h:1470
↓ 2 callersFunctionfind_include_path
Find header files for C compilation.
python/tvm_ffi/libinfo.py:87
↓ 2 callersMethodfreeRegion
addons/tvm_ffi_orcjit/src/ffi/orcjit_slab.cc:423
↓ 2 callersMethodgen_methods
Generate method definitions for this object.
python/tvm_ffi/stub/utils.py:274
↓ 2 callersFunctiongenerate_ffi_api
Generate the initial FFI API stub code for a given module.
python/tvm_ffi/stub/codegen.py:243
↓ 2 callersFunctiongenerate_import_section
Generate import statements for the types used in the stub. It processes: import-section
python/tvm_ffi/stub/codegen.py:147
↓ 2 callersFunctiongenerate_init
Generate the `__init__.py` file for the `tvm_ffi` package.
python/tvm_ffi/stub/codegen.py:304
↓ 2 callersFunctiongenerate_object
Generate a class definition for an object type. It processes: object/${type_key}
python/tvm_ffi/stub/codegen.py:97
↓ 2 callersFunctionget_attr
Get an attribute by name from a derive input # Arguments `derive_input` - The derive input to get the attribute from `name` - The name of the attribu
rust/tvm-ffi-macros/src/utils.rs:41
↓ 2 callersFunctionget_tvm_ffi_crate
Get the tvm-rt crate name \return The tvm-rt crate name
rust/tvm-ffi-macros/src/utils.rs:25
↓ 2 callersFunctiongetitem_helper
Implement a pythonic __getitem__ helper. Parameters ---------- obj The original object elem_getter A simple function
python/tvm_ffi/container.py:121
↓ 2 callersMethodimport_module
Add module to the import list of current one. Parameters ---------- module The other module.
python/tvm_ffi/module.py:279
↓ 2 callersMethodinto_iter
(self)
rust/tvm-ffi/src/collections/array.rs:221
↓ 2 callersMethodisValid
! \brief True iff the reservation succeeded. */
addons/tvm_ffi_orcjit/src/ffi/orcjit_slab.h:176
↓ 2 callersMethodis_high
(self)
tests/python/test_dataclass_enum.py:271
↓ 2 callersMethoditem_size
(&self)
rust/tvm-ffi/src/collections/tensor.rs:68
↓ 2 callersMethodlink_order_base_obj
Return path prefix for test_link_order_base object.
addons/tvm_ffi_orcjit/tests/test_basic.py:103
↓ 2 callersFunctionlist_global_func_names
Get list of global functions registered. Returns ------- names List of global functions names.
python/tvm_ffi/registry.py:223
↓ 2 callersFunctionload_torch_get_current_cuda_stream
Create a faster get_current_cuda_stream for torch through cpp extension.
tests/scripts/benchmark_dlpack.py:356
↓ 2 callersMethodnext
(&mut self)
rust/tvm-ffi/src/collections/array.rs:206
↓ 2 callersFunctionobject_info_from_type_key
Construct an `ObjectInfo` from an object type key.
python/tvm_ffi/stub/lib_state.py:34
↓ 2 callersFunctionoptional_tensor_view_has_value
(_0: Tensor | None, /)
python/tvm_ffi/testing/_ffi_api.py:55
↓ 2 callersFunctionparse_env_flags
(env_var_name: str)
python/tvm_ffi/utils/_build_optional_torch_c_dlpack.py:576
↓ 2 callersFunctionprint_speed
Print benchmark result in a formatted way. Parameters ---------- name : str Name of the benchmark time_per_call : float
examples/cubin_launcher/benchmark_overhead.py:71
↓ 2 callersMethodprotectPages
addons/tvm_ffi_orcjit/src/ffi/orcjit_slab.cc:201
↓ 2 callersMethodpush_back
mutation in std::vector, implements copy-on-write ! * \brief push a new item to the back of the list * \param item The item to be pushed. */
include/tvm/ffi/container/array.h:426
↓ 2 callersFunctionpy_class
Register a Python-defined FFI class with dataclass-style semantics. Can be used as: .. code-block:: python @py_class # bare decora
python/tvm_ffi/dataclasses/py_class.py:630
↓ 2 callersMethodreload
Reload the code blocks from disk while preserving original `lines`.
python/tvm_ffi/stub/file_utils.py:223
↓ 2 callersFunctionrun_bench
(device: str)
tests/scripts/benchmark_dlpack.py:74
↓ 2 callersFunctionrun_check
Must run in a separate function to ensure deletion happens before mod unloads. When a module returns an object, the object deleter address is
tests/python/test_load_inline.py:245
↓ 2 callersMethodsame_as
include/tvm/ffi/container/variant.h:46
↓ 2 callersFunctiontarget_func
(*args: Any)
tests/scripts/benchmark_kwargs_wrapper.py:32
↓ 2 callersMethodtest_vector
Vector dtype roundtrips correctly.
tests/python/test_serialization.py:203
↓ 2 callersFunctiontvm_ffi_nop_autodlpack_from_dltensor_test_wrapper
Measures overhead of running dlpack via auto convert by directly take test wrapper as inputs. This effectively measure DLPack exchange in tvm ffi.
tests/scripts/benchmark_dlpack.py:286
↓ 2 callersFunctiontvm_ffi_nop_autodlpack_from_test_tensor_namedtuple
Measures overhead of running dlpack via auto convert by directly take test wrapper as inputs. This effectively measure DLPack exchange in tvm ffi.
tests/scripts/benchmark_dlpack.py:301
↓ 2 callersFunctionty_map
(name: str)
tests/python/test_stubgen.py:162
↓ 2 callersMethodwith_lanes
Create a new dtype with the given number of lanes. Parameters ---------- lanes The number of lanes for the result
python/tvm_ffi/_dtype.py:146
↓ 1 callersFunctionAccessPathRegisterReflection
src/ffi/extra/reflection_extra.cc:146
↓ 1 callersFunctionAccessStepRegisterReflection
src/ffi/extra/reflection_extra.cc:103
↓ 1 callersMethodAdd
src/ffi/extra/module.cc:41
↓ 1 callersMethodAddPendingDeinitializer
addons/tvm_ffi_orcjit/src/ffi/orcjit_session.cc:286
↓ 1 callersMethodAddPendingInitializer
addons/tvm_ffi_orcjit/src/ffi/orcjit_session.cc:281
↓ 1 callersMethodAnyView
! \brief Default constructor */
include/tvm/ffi/any.h:74
↓ 1 callersFunctionAny_AnyView_GetInt
[Any_AnyView.GetInt_Float.begin]
examples/abi_overview/example_code.c:89
↓ 1 callersFunctionAtomicLoadHandleAcquire
src/ffi/init_once.cc:36
↓ 1 callersFunctionAtomicStoreHandleRelease
src/ffi/init_once.cc:50
← previousnext →501–600 of 5,333, ranked by callers