kernel

Attributes

Classes

Module Contents

GENERIC_TEMPLATE = Multiline-String
Show Value
"""
import warp as wp

@wp.kernel
def evaluate_{domain}_energy(
    state_vector: wp.array(dtype=wp.vec3),  # Position/Velocity mixed
    properties: wp.array(dtype=float),      # Mass/Pressure/Signal
    loss: wp.array(dtype=float)
):
    tid = wp.tid()

    # Unpack State
    sv = state_vector[tid]
    prop = properties[tid]

    # --- AI MATH INJECTION ---
    # Logic: {logic}
    score = {logic}
    # -------------------------

    # Safety Clipping (From Pydantic)
    safe_score = wp.clamp(score, {min_val}, {max_val})

    wp.atomic_add(loss, 0, safe_score)
"""
class KernelFactory
static compile(spec)