tilus.ir.layout.auto_local_spatial

tilus.ir.layout.auto_local_spatial

tilus.ir.layout.auto_local_spatial(num_threads, shape)[source]

Create a local(…).spatial(…) layout

This function automatically determines a composition of the local and spatial layouts, based on the number of threads and the shape of the composed layout.

Parameters:
  • num_threads (int) – The number of threads to be used for the spatial layout. This should be a positive integer.

  • shape (Sequence[int]) – The shape of the composed layout. Each entry in shape must be a positive constant integer.

Returns:

ret – The layout that is a composition of local and spatial layouts.

Return type:

RegisterLayout