Matt
Migrated from model repo Rocketknight1/esmfold2-trimul-kernel
c75f6f8
Raw
History Blame
725 Bytes
# 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