activation.py File Reference

Namespaces
namespace	iron
namespace	iron.kernels
namespace	iron.kernels.activation

Functions
ExternalFunction	iron.kernels.activation._create_lut_kernel (str func_name, str kernel_filename, list arg_types, list[str]|None compile_flags=None)
ExternalFunction	iron.kernels.activation._bf16_lut_factory (str factory_name, str func_name, str kernel_filename, int tile_size, int arg_arity)
ExternalFunction	iron.kernels.activation.softmax (int tile_size=1024)
ExternalFunction	iron.kernels.activation.gelu (int tile_size=1024)
ExternalFunction	iron.kernels.activation.silu (int tile_size=1024)
ExternalFunction	iron.kernels.activation.swiglu (int tile_size=1024)
ExternalFunction	iron.kernels.activation.bf16_exp (int tile_size=1024)
	iron.kernels.activation.relu_ref (x)
	iron.kernels.activation.silu_ref (x)
	iron.kernels.activation.gelu_ref (x)
	iron.kernels.activation.bf16_exp_ref (x)
	iron.kernels.activation.softmax_ref (x, *int tile_size=1024)

Variables
int	iron.kernels.activation._LUT_FIXED_TILE = 1024