# kernel-builder config for the ESMFold2 triangle-multiplication kernel. # # This is a pure-Triton kernel (no C++/CUDA to compile), so it builds as a # "universal" Torch extension: kernel-builder packages the Python+Triton sources # under build/torch-universal/ and the `kernels` library JIT-compiles the Triton # at runtime. Verify field names against the kernel-builder version you build with # (https://github.com/huggingface/kernel-builder) — the universal-build schema has # shifted across releases. [general] # Must match the repo-derived package name: repo "esmfold2-trimul-kernel" # -> kernels uses package_name = "esmfold2_trimul_kernel" (hyphens -> underscores). name = "esmfold2_trimul_kernel" universal = true