View page source

warp.set_module_options#

warp.set_module_options(options, module=None)[source]#

Set options for the current module.

Options can be used to control runtime compilation and code-generation for the current module individually. Available options are listed below.

  • mode: The compilation mode to use, can be “debug”, or “release”, defaults to the value of warp.config.mode.

  • max_unroll: The maximum fixed-size loop to unroll, defaults to the value of warp.config.max_unroll.

  • enable_mathdx_gemm: Use libmathdx (cuBLASDx) for tile_matmul on GPU. If None (the default), defers to warp.config.enable_mathdx_gemm at compile time.

  • block_dim: The default number of threads to assign to each block

Parameters:

  • options (dict[str, Any]) – Set of key-value option pairs

  • module (Any)