brainevent.load_cuda_dir

Contents

brainevent.load_cuda_dir#

brainevent.load_cuda_dir(directory, functions=None, *, name=None, file_patterns=None, **kwargs)[source]#

Compile all CUDA files in a directory and load the resulting module.

Parameters:
  • directory (str | Path) – Directory containing .cu / .cuh files.

  • functions (dict[str, list[str]] | None) – Function name → arg_spec mapping. If None, discovered from // @BE annotations.

  • name (str | None) – Module name. Defaults to the directory name.

  • file_patterns (list[str] | None) – Glob patterns for source files (default: ["*.cu"]).

  • **kwargs – Forwarded to load_cuda_inline.

Return type:

CompiledModule