brainevent.register_ffi_target

brainevent.register_ffi_target#

brainevent.register_ffi_target(target_name, module, func_name, *, platform='CUDA')[source]#

Register a compiled function as a JAX FFI target.

After registration, the function can be invoked inside @jax.jit via jax.ffi.ffi_call(target_name, ...).

Parameters:

  • target_name (str) – Globally unique FFI target identifier.

  • module (CompiledModule) – The loaded module containing the function.

  • func_name (str) – Function name within the module.

  • platform (str) – Target platform ("CUDA" or "cpu").

Return type:

None