"""Context-parallel stubs. Pass ``cp_context=None`` (the default).""" class FLACPContext: cu_seqlens = None cu_seqlens_cpu = None def build_cp_context(*args, **kwargs): raise RuntimeError("Context parallel is not included in the vendored KDA kernels") __all__ = ["FLACPContext", "build_cp_context"]