Module `Tolk.Search`

Beam search kernel optimiser.

Explores the space of kernel optimisations by compiling, timing, and selecting the best candidates over multiple rounds.

val beam_search : 
  ?allow_test_size:bool ->
  ?disable_cache:bool ->
  Postrange.t ->
  Device.Buffer.t list ->
  int ->
  Device.t ->
  Postrange.t

beam_search s rawbufs amt device optimises scheduler s using beam search with beam width amt.

allow_test_size (default true) scales down global dimensions during timing to stay within hardware limits.
disable_cache (default from IGNORE_BEAM_CACHE env) bypasses on-disk cache reads. Successful searches still update the cache when caching is enabled.

Returns the best scheduler found.