eltwise.py File Reference

Namespaces
namespace	iron
namespace	iron.kernels
namespace	iron.kernels.eltwise

Functions
ExternalFunction	iron.kernels.eltwise._eltwise_bf16_kernel (str op, int tile_size, dtype, bool vectorized)
ExternalFunction	iron.kernels.eltwise.passthrough (int tile_size=4096, dtype=np.int32)
ExternalFunction	iron.kernels.eltwise.scale (int tile_size=1024, dtype=np.int32, bool vectorized=True, bool use_chess=False)
ExternalFunction	iron.kernels.eltwise.add (int tile_size=1024, dtype=bfloat16, bool vectorized=True)
ExternalFunction	iron.kernels.eltwise.mul (int tile_size=1024, dtype=bfloat16, bool vectorized=True)
ExternalFunction	iron.kernels.eltwise.relu (int tile_size=1024)

Variables
int	iron.kernels.eltwise._ELTWISE_FIXED_TILE = 1024
int	iron.kernels.eltwise._RELU_FIXED_TILE = 1024