File cubin_launcher.h

Contents

File cubin_launcher.h#

↰ Parent directory (tvm/ffi/extra/cuda)

CUDA CUBIN launcher utility for loading and executing CUDA kernels.

Definition (`tvm/ffi/extra/cuda/cubin_launcher.h`)#

Program Listing for File cubin_launcher.h

Detailed Description#

This header provides a lightweight C++ wrapper around CUDA Runtime API for loading CUBIN modules and launching kernels. It supports:

Loading CUBIN from memory (embedded data)
Multi-GPU execution using CUDA primary contexts
Kernel parameter management and launch configuration

Includes#

cstdint
cstring
cuda_runtime.h
tvm/ffi/error.h (File error.h)
tvm/ffi/extra/c_env_api.h (File c_env_api.h)
tvm/ffi/extra/cuda/base.h (File base.h)
tvm/ffi/string.h (File string.h)

Namespaces#

Classes#

Defines#