Skip to content

Comments

feat: enable bundling cutlass in build#298

Open
drbh wants to merge 1 commit intomainfrom
support-bundling-deps
Open

feat: enable bundling cutlass in build#298
drbh wants to merge 1 commit intomainfrom
support-bundling-deps

Conversation

@drbh
Copy link
Collaborator

@drbh drbh commented Feb 20, 2026

This is an experimental feature that adds a bundle-dep-includes attribute to the build toml that enables the inclusion of cutlass headers in the build artifacts.

This feature enables the deep-gemm kernel to be built, since the kernel does JIT compilation using the cutlass and cute library at runtime.

Initially I opted to vendor the dependencies in the kernel source but that adds hundreds of files that we already have available within the builder itself, this approach reuses the existing files and writes them into the build as shown below

└── torch29-cxx11-cu130-x86_64-linux
    ├── deep_gemm
    │   └── __init__.py
    ├── _deep_gemm_cuda_7f1079e_dirty.abi3.so
    ├── include
    │   ├── cute
    │   ├── cutlass
    │   └── deep_gemm
    ├── __init__.py
    ├── metadata.json
    ├── _ops.py
    └── utils
        ├── __init__.py
        └── math.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant