MCPcopy Create free account

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

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

↓ 1 callersFunctionc_class
Register a C++ FFI class and install structural dunder methods. Combines :func:`~tvm_ffi.register_object` with structural comparison, hashing
python/tvm_ffi/dataclasses/c_class.py:64
↓ 1 callersMethodcall_global_obj
Return path prefix for test_call_global object.
addons/tvm_ffi_orcjit/tests/test_basic.py:95
↓ 1 callersMethodcause_chain
! * \brief Get the cause chain of the error object. * \return The cause chain of the error object. */
include/tvm/ffi/error.h:199
↓ 1 callersFunctioncheck0
()
tests/python/test_function.py:197
↓ 1 callersFunctioncheck1
()
tests/python/test_function.py:204
↓ 1 callersFunctioncheck_asf_copyright
(fname: str)
tests/lint/check_file_type.py:173
↓ 1 callersFunctioncheck_header
Check header status of file without modifying it.
tests/lint/check_asf_header.py:214
↓ 1 callersFunctioncheck_tensor
()
tests/python/test_function.py:93
↓ 1 callersMethodclear
Remove all elements from the dict.
python/tvm_ffi/container.py:767
↓ 1 callersMethodclear
! \brief Release reference to all the elements */
include/tvm/ffi/container/array.h:539
↓ 1 callersMethodclear
! * \brief clear all entries */
include/tvm/ffi/container/map_base.h:1127
↓ 1 callersMethodclear
! \brief Release reference to all the elements */
include/tvm/ffi/container/list.h:424
↓ 1 callersMethodclearFreeSlabs
addons/tvm_ffi_orcjit/src/ffi/orcjit_memory_manager.cc:149
↓ 1 callersFunctioncollect_files
Collect all files from the given paths and parse them into FileInfo objects.
python/tvm_ffi/stub/file_utils.py:230
↓ 1 callersFunctioncollect_files
Collect all files that need header checking from git.
tests/lint/check_asf_header.py:246
↓ 1 callersFunctioncollect_global_funcs
Collect global functions from TVM FFI's global registry.
python/tvm_ffi/stub/lib_state.py:41
↓ 1 callersFunctioncollect_type_keys
Collect registered object type keys from TVM FFI's global registry.
python/tvm_ffi/stub/lib_state.py:59
↓ 1 callersMethodcompare
! * \brief Constructor. * \param include Whether the field should participate in comparison. */
include/tvm/ffi/reflection/registry.h:292
↓ 1 callersFunctioncopyright_line
(line: str)
tests/lint/check_file_type.py:161
↓ 1 callersMethodcreate
( tb: types.TracebackType | None, frame: types.FrameType, lineno: int )
python/tvm_ffi/error.py:130
↓ 1 callersMethoddeallocateOne
addons/tvm_ffi_orcjit/src/ffi/orcjit_slab.cc:727
↓ 1 callersFunctiondecorator
(cls: _T)
python/tvm_ffi/dataclasses/py_class.py:793
↓ 1 callersMethoddefault
(cls)
tests/python/test_dataclass_enum.py:275
↓ 1 callersMethodderef
(&self)
rust/tvm-ffi/src/collections/shape.rs:113
↓ 1 callersFunctiondevice
Construct a TVM FFI device with given device type and index. Parameters ---------- device_type: str or int The device type or nam
python/tvm_ffi/_tensor.py:81
↓ 1 callersMethoddoubled
(self)
tests/python/test_typed_method.py:167
↓ 1 callersFunctionensure_compile_commands
(build_dir: Path)
tests/lint/clang_tidy_precommit.py:81
↓ 1 callersFunctionerror_fn1
(flag: bool)
rust/tvm-ffi/tests/test_error.rs:47
↓ 1 callersMethoderror_obj
Return path prefix for test_error object.
addons/tvm_ffi_orcjit/tests/test_basic.py:111
↓ 1 callersFunctionfib
fibonacci: recursive fibonacci */
addons/tvm_ffi_orcjit/examples/quick-start/add_c.c:48
↓ 1 callersFunctionfilename_allowed
Check if name is allowed by the current policy. Paramaters ---------- name : str Input name Returns ------- allowed
tests/lint/check_file_type.py:131
↓ 1 callersFunctionfilter_files
(files: Sequence[str])
tests/lint/clang_tidy_precommit.py:65
↓ 1 callersMethodfind
include/tvm/ffi/container/map_base.h:1474
↓ 1 callersMethodfind
! \return find the key and returns the associated iterator */
include/tvm/ffi/container/dict.h:248
↓ 1 callersFunctionfind_libtvm_ffi
Find libtvm_ffi. Returns ------- path The full path to the located library.
python/tvm_ffi/libinfo.py:36
↓ 1 callersFunctionfind_python_helper_include_path
Find header files for C compilation.
python/tvm_ffi/libinfo.py:124
↓ 1 callersFunctionfooter_html
Generate HTML for the documentation footer.
docs/conf.py:434
↓ 1 callersFunctionfor_each
include/tvm/ffi/base_details.h:238
↓ 1 callersMethodfree_data
(&mut self, tensor: &DLTensor)
rust/tvm-ffi/src/collections/tensor.rs:336
↓ 1 callersMethodfrom_dlpack_data_type
Create a dtype from a DLPack data type tuple. Parameters ---------- dltype_data_type The DLPack data type tuple `
python/tvm_ffi/_dtype.py:100
↓ 1 callersMethodfrom_type_info
Construct an `ObjectInfo` from a `TypeInfo` instance.
python/tvm_ffi/stub/utils.py:210
↓ 1 callersMethodgen
()
tests/python/test_type_converter.py:414
↓ 1 callersMethodgen_ffi_init
Generate a ``__ffi_init__`` stub when it's not already in TypeMethod. For types whose ``__ffi_init__`` is auto-generated by ``RegisterFFIInit
python/tvm_ffi/stub/utils.py:312
↓ 1 callersMethodgen_init
Generate an ``__init__`` stub from init-eligible field metadata.
python/tvm_ffi/stub/utils.py:327
↓ 1 callersFunctiongenerate_cubin
Define CUDA kernels and compile them to a CUBIN file. The kernels are named `add_one` and `mul_two` and compute y[i] = x[i] + 1 and y[i] = x[i] *
examples/cubin_launcher/example_nvrtc_cubin.py:41
↓ 1 callersFunctiongenerate_cubin
Compile the empty kernel to CUBIN. Returns ------- bytes Compiled CUBIN bytes
examples/cubin_launcher/benchmark_overhead.py:121
↓ 1 callersFunctiongenerate_cubin
Define a Triton kernel in-process and compile it to a CUBIN file. The kernel is named `square_kernel` and computes y[i] = x[i] * x[i]. Retur
examples/cubin_launcher/example_triton_cubin.py:43
↓ 1 callersFunctiongenerate_example_lib
()
rust/tvm-ffi/build.rs:22
↓ 1 callersFunctiongenerate_export
Generate an `__all__` variable for the given names.
python/tvm_ffi/stub/codegen.py:223
↓ 1 callersMethodget
Retrieves an item at the given index.
rust/tvm-ffi/src/collections/array.rs:136
↓ 1 callersMethodget
include/tvm/ffi/container/variant.h:191
↓ 1 callersFunctionget_cpu_name
Get the name of the CPU.
examples/cubin_launcher/benchmark_overhead.py:48
↓ 1 callersFunctionget_first_structural_mismatch
Like structural_equal(), but returns the AccessPath pair of the first detected mismatch. Parameters ---------- lhs The left opera
python/tvm_ffi/structural.py:134
↓ 1 callersMethodget_function
Get a function from the module by name. # Arguments `name` - The name of the function to retrieve # Returns `Result<Function>` - A `Function` instan
rust/tvm-ffi/src/extra/module.rs:69
↓ 1 callersFunctionget_git_files
Get list of files tracked by git.
tests/lint/check_asf_header.py:185
↓ 1 callersFunctionget_torch_include_paths
Get the include paths for building with torch.
python/tvm_ffi/utils/_build_optional_torch_c_dlpack.py:719
↓ 1 callersFunctionget_val
Helper to extract the first float value from a Tensor
rust/tvm-ffi/tests/test_array.rs:33
↓ 1 callersFunctionhidden_helper_add
Hidden visibility: caller uses ADRP+ADD instead of GOT */
addons/tvm_ffi_orcjit/tests/sources/c/test_hidden_helper.c:33
↓ 1 callersFunctionhidden_helper_add
Hidden visibility: caller uses ADRP+ADD instead of GOT
addons/tvm_ffi_orcjit/tests/sources/cc/test_hidden_helper.cc:31
↓ 1 callersMethodinsert
! * \brief Insert an element into the given position * \param position An iterator pointing to the insertion point * \param val The element t
include/tvm/ffi/container/array.h:446
↓ 1 callersFunctioninvoke_testing_add1
tests/cpp/test_function.cc:286
↓ 1 callersFunctionisAbsoluteEdge
addons/tvm_ffi_orcjit/src/ffi/orcjit_slab.cc:79
↓ 1 callersMethodisReclaimable
! * \brief True iff this slab has ever hosted a FinalizedAlloc and * currently has zero live ones. * * Used by `SlabPoolMemoryMana
addons/tvm_ffi_orcjit/src/ffi/orcjit_slab.h:263
↓ 1 callersMethodkind
Get the kind of the error # Returns The kind of the error
rust/tvm-ffi/src/error.rs:120
↓ 1 callersMethodkind
src/ffi/extra/library_module.cc:40
↓ 1 callersFunctionload_cubin_module
Load CUBIN kernel through TVM-FFI. Parameters ---------- cubin_bytes : bytes Compiled CUBIN bytes Returns ------- Mo
examples/cubin_launcher/benchmark_overhead.py:143
↓ 1 callersFunctionload_torch_c_dlpack_extension
()
python/tvm_ffi/_optional_torch_c_dlpack.py:103
↓ 1 callersFunctionload_torch_c_dlpack_extension
Load the torch c dlpack extension based on torch version.
addons/torch_c_dlpack_ext/torch_c_dlpack_ext/core.py:53
↓ 1 callersFunctionmain
Main entry point for the embed_cubin utility.
python/tvm_ffi/utils/embed_cubin.py:306
↓ 1 callersFunctionmain
Build the torch c dlpack extension.
python/tvm_ffi/utils/_build_optional_torch_c_dlpack.py:729
↓ 1 callersFunctionmain
Generate the FFI library for examples.
rust/tvm-ffi/scripts/generate_example_lib.py:26
↓ 1 callersFunctionmain
()
tests/lint/check_file_type.py:193
↓ 1 callersFunctionmain
()
tests/lint/check_version.py:160
↓ 1 callersFunctionmain
(argv: Sequence[str] | None = None)
tests/lint/clang_tidy_precommit.py:140
↓ 1 callersFunctionmain
()
tests/lint/check_asf_header.py:326
↓ 1 callersFunctionmain
()
tests/scripts/benchmark_pycallback.py:91
↓ 1 callersFunctionmain
()
tests/scripts/benchmark_dlpack.py:414
↓ 1 callersFunctionmain
Build, load, and run inline CPU/CUDA functions.
examples/inline_module/main.py:24
↓ 1 callersFunctionmain
Compile and launch CUDA kernel through NVRTC -> TVM-FFI.
examples/cubin_launcher/example_nvrtc_cubin.py:236
↓ 1 callersFunctionmain
Main benchmark entry point.
examples/cubin_launcher/benchmark_overhead.py:271
↓ 1 callersFunctionmain
Load and launch Triton kernel through TVM-FFI.
examples/cubin_launcher/example_triton_cubin.py:169
↓ 1 callersFunctionmain
Test the lib_embedded.so library with embedded CUBIN.
examples/cubin_launcher/embedded_cubin/cpp_embed/main.py:31
↓ 1 callersFunctionmain
Test the lib_embedded.so library with embedded CUBIN.
examples/cubin_launcher/embedded_cubin/embed_with_tvm_ffi/main.py:31
↓ 1 callersFunctionmain
Test the lib_embedded.so library with embedded CUBIN.
examples/cubin_launcher/embedded_cubin/include_bin2c/main.py:31
↓ 1 callersFunctionmain
Test the lib_dynamic.so library with dynamic CUBIN loading.
examples/cubin_launcher/dynamic_cubin/main.py:31
↓ 1 callersFunctionmain
Run the quick start example.
addons/tvm_ffi_orcjit/examples/quick-start/run.py:120
↓ 1 callersMethodmap_item
Create an access path to the item of the current map. Parameters ---------- key The key of the item to access
python/tvm_ffi/access_path.py:217
↓ 1 callersMethodmap_item_missing
Create an access path that indicate a map item is missing. Parameters ---------- key The key of the item to acces
python/tvm_ffi/access_path.py:233
↓ 1 callersFunctionmay_create_cyclic_reference
()
tests/python/test_error.py:141
↓ 1 callersFunctionmethod
Mark a ``@py_class`` method for FFI TypeMethod registration. Decorate any staticmethod or plain instance method on a ``@py_class`` body to ha
python/tvm_ffi/dataclasses/py_class.py:240
↓ 1 callersMethodnoteAllocated
! * \brief Record that a new FinalizedAlloc was published from this slab. * Called by `InFlightAlloc::finalize` just before returning the
addons/tvm_ffi_orcjit/src/ffi/orcjit_slab.h:243
↓ 1 callersMethodoperator->
include/tvm/ffi/container/map_base.h:1433
↓ 1 callersMethodoperator[]
! * \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:275
↓ 1 callersMethodoperator[]
! * \brief Read element from map. * \param key The key * \return the corresonding element. */
include/tvm/ffi/container/map.h:205
↓ 1 callersFunctionparse_args
(argv: Sequence[str])
tests/lint/clang_tidy_precommit.py:30
↓ 1 callersFunctionpatch_torch_cuda_stream_protocol
Load the torch cuda stream protocol for older versions of torch.
python/tvm_ffi/_optional_torch_c_dlpack.py:188
↓ 1 callersFunctionpath_walk
Compat wrapper for Path.walk (3.12+) with a fallback for < 3.12.
python/tvm_ffi/stub/file_utils.py:266
↓ 1 callersFunctionpopulate_object_table
(num_classes: int)
tests/scripts/benchmark_dlpack.py:407
↓ 1 callersFunctionprint_error
(name: str, error: Any)
tests/scripts/benchmark_dlpack.py:67
↓ 1 callersFunctionprint_speed
(name: str, speed: float)
tests/scripts/benchmark_unpack_dataclass.py:28
← previousnext →901–1,000 of 5,333, ranked by callers