Register the error class name with the FFI core.

Register the global function with the FFI core.

Register object type. Parameters ---------- type_key The type key of the node. It requires ``type_key`` to be registered already

Remove a global function by name. Parameters ---------- name The name of the global function. Examples -------- .. c

! * \brief Make sure the list has the capacity of at least n * \param n lower bound of the capacity */

! \brief reset the content of ptr to be nullptr */

! \brief reset the content of ptr to be nullptr */

Invoke add_one from the extension and print the result.

Invoke add_two from the extension and print the result.

Run overhead benchmarks for both Triton and TVM-FFI. Parameters ---------- cubin_bytes : bytes Compiled CUBIN bytes num_calls

Invoke IntPair from the extension to demonstrate object handling.

Invoke raise_error from the extension to demonstrate error handling.

! * \brief Comparator * \param other Another object ref. * \return the compare result. */

! * \brief Get the shape of the Tensor. * \return The shape of the Tensor. */

! * \brief Get the shape of the Tensor. * \return The shape of the Tensor. */

Check if file should be skipped based on SKIP_LIST.

tvm_ffi.Map → canonical origin 'Map'.

Collect ObjectInfo objects for type keys, topologically sorted by inheritance.

Overhead of tvm FFI python call via calling a NOP. testing.nop is defined in c++ and do nothing.

Measures overhead of running dlpack via auto convert by directly take numpy.ndarray as inputs.

Run dlpack conversion + tvm_ffi.nop. Measures overhead of running dlpack for each args then invoke

Run dlpack conversion + tvm_ffi.nop. Measures overhead of running dlpack for each args then invoke

Measures overhead of running dlpack for each args then invoke but uses the legacy torch.utils.dlpack.to_dlpack API. This helps to measure pos

Run dlpack conversion + tvm_ffi.nop. Measures overhead of running dlpack for each args then invoke

! \return The internal runtime type index of the object. */

Update the dict from a mapping.

Update the LD_LIBRARY_PATH environment variable so cargo run/test can directly pick it up note that it won't always work later consumption of the libr

Update the LD_LIBRARY_PATH environment variable so cargo run/test can directly pick it up note that it won't always work later consumption of the libr

Load and test CUBIN kernels through TVM-FFI. Parameters ---------- cubin_bytes : bytes Compiled CUBIN bytes Returns ----

Load and test Triton CUBIN kernel through TVM-FFI. Parameters ---------- cubin_bytes : bytes Compiled CUBIN bytes Returns

Call `f` with the device stream temporarily set to `stream`. # Safety `stream` must be a valid stream handle for the given device, or null.

Worker function that tries to acquire lock and increment counter.

! * \brief Destructor */

default constructor to enable auto-serialization

! * \brief Constructor * \param kind The kind of the access step. * \param key The key of the access step. */

default constructor to enable auto-serialization

Some methods to exercise RegisterMethod

! * \brief Launch add_one_cuda kernel on input tensor. * \param x Input tensor (float32, 1D) * \param y Output tensor (float32, 1D, same shape as x

! * \brief Launch add_one_cuda kernel on input tensor. * \param x Input tensor (float32, 1D) * \param y Output tensor (float32, 1D, same shape as x

! * \brief Launch add_one_cuda kernel on input tensor. * \param x Input tensor (float32, 1D) * \param y Output tensor (float32, 1D, same shape as x

! * \brief Launch add_one_cuda kernel on input tensor. * \param x Input tensor (float32, 1D) * \param y Output tensor (float32, 1D, same shape as x

! \brief Perform vector add one: y = x + 1 (1-D float32) */

[tvm_ffi_abi.begin]

! * \brief Default constructor */

[Any_AnyView.Destroy.begin]

[Any_AnyView.FromObjectPtr.begin]

[Any_AnyView.GetDLTensor.begin]

[Any_AnyView.GetObject.begin]

! * \brief Move metadata into `FieldInfoBuilder` * \param info The field info builder. */

! * \brief Apply the default value to the field info * \param info The field info. */

! * \brief Apply the default factory to the field info * \param info The field info. */

! * \brief Apply the field flag to the field info * \param info The field info. */

! * \brief Apply the compare flag to the field info. * \param info The field info. */

! * \brief Apply the hash flag to the field info. * \param info The field info. */

! * \brief Apply the kw_only flag to the field info. * \param info The field info. */

! * \brief Apply the init flag to the field info. * \param info The field info. */

! * \brief Create an access step for a array item access. * \param index The index of the array item to access. * \return The access step.

! * \brief Create an access step for a array item missing access. * \param index The index of the array item to access. * \return The access

! * \brief Attach a field flag to the field * \param flag The flag to be set */

! * \brief Create an access step for a object attribute access. * \param field_name The name of the field to access. * \return The access ste

! * \brief Create an access step for a object attribute missing access. * \param field_name The name of the field to access. * \return The ac