Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion 3rdparty/tvm-ffi
Submodule tvm-ffi updated 69 files
+3 −0 .gitignore
+4 −0 CMakeLists.txt
+189 −0 KEYS
+1 −1 addons/torch_c_dlpack_ext/pyproject.toml
+6 −1 addons/torch_c_dlpack_ext/torch_c_dlpack_ext/core.py
+219 −0 cmake/Utils/EmbedCubin.cmake
+1 −0 cmake/tvm_ffi-config.cmake
+2 −0 docs/conf.py
+2 −1 docs/get_started/quickstart.rst
+7 −3 docs/guides/compiler_integration.md
+45 −2 docs/guides/cpp_guide.md
+422 −0 docs/guides/cubin_launcher.rst
+186 −0 docs/guides/kernel_library_guide.rst
+2 −0 docs/index.rst
+7 −0 docs/reference/python/index.rst
+176 −0 examples/cubin_launcher/README.md
+311 −0 examples/cubin_launcher/benchmark_overhead.py
+62 −0 examples/cubin_launcher/dynamic_cubin/CMakeLists.txt
+122 −0 examples/cubin_launcher/dynamic_cubin/main.py
+54 −0 examples/cubin_launcher/dynamic_cubin/src/kernel.cu
+132 −0 examples/cubin_launcher/dynamic_cubin/src/lib_dynamic.cc
+69 −0 examples/cubin_launcher/embedded_cubin/CMakeLists.txt
+112 −0 examples/cubin_launcher/embedded_cubin/main.py
+54 −0 examples/cubin_launcher/embedded_cubin/src/kernel.cu
+115 −0 examples/cubin_launcher/embedded_cubin/src/lib_embedded.cc
+268 −0 examples/cubin_launcher/example_nvrtc_cubin.py
+208 −0 examples/cubin_launcher/example_triton_cubin.py
+1 −1 include/tvm/ffi/c_api.h
+49 −0 include/tvm/ffi/container/tensor.h
+54 −0 include/tvm/ffi/extra/cuda/base.h
+604 −0 include/tvm/ffi/extra/cuda/cubin_launcher.h
+74 −0 include/tvm/ffi/extra/cuda/device_guard.h
+25 −6 include/tvm/ffi/extra/module.h
+166 −49 include/tvm/ffi/function.h
+5 −1 pyproject.toml
+22 −0 python/tvm_ffi/__init__.py
+3 −1 python/tvm_ffi/_convert.py
+28 −0 python/tvm_ffi/_dtype.py
+51 −8 python/tvm_ffi/_ffi_api.py
+52 −9 python/tvm_ffi/_optional_torch_c_dlpack.py
+31 −23 python/tvm_ffi/access_path.py
+1 −0 python/tvm_ffi/core.pyi
+158 −45 python/tvm_ffi/cpp/extension.py
+171 −0 python/tvm_ffi/cpp/nvrtc.py
+2 −2 python/tvm_ffi/cython/function.pxi
+18 −0 python/tvm_ffi/cython/tensor.pxi
+111 −11 python/tvm_ffi/module.py
+41 −0 python/tvm_ffi/stub/analysis.py
+225 −0 python/tvm_ffi/stub/cli.py
+137 −0 python/tvm_ffi/stub/codegen.py
+60 −0 python/tvm_ffi/stub/consts.py
+243 −0 python/tvm_ffi/stub/file_utils.py
+0 −528 python/tvm_ffi/stub/stubgen.py
+151 −0 python/tvm_ffi/stub/utils.py
+49 −44 python/tvm_ffi/testing.py
+103 −51 python/tvm_ffi/utils/_build_optional_torch_c_dlpack.py
+405 −0 python/tvm_ffi/utils/embed_cubin.py
+24 −0 src/ffi/extra/library_module.cc
+139 −53 src/ffi/testing/testing.cc
+15 −0 tests/cpp/test_function.cc
+45 −0 tests/cpp/test_metadata.cc
+2 −0 tests/python/test_build.cc
+276 −0 tests/python/test_build.py
+321 −0 tests/python/test_cubin_launcher.py
+17 −3 tests/python/test_dlpack_exchange_api.py
+22 −0 tests/python/test_dtype.py
+323 −144 tests/python/test_stubgen.py
+11 −0 tests/python/test_tensor.py
+255 −0 tests/python/utils/test_embed_cubin.py
Loading