defines.h File Reference

Go to the source code of this file.

Macros
#define	MTL_CONST
#define	instantiate_kernel(name, func, ...)

Variables
static constexpr int	MAX_REDUCE_SPECIALIZED_DIMS = 4
static constexpr int	REDUCE_N_READS = 16
static constexpr int	SOFTMAX_N_READS = 4
static constexpr int	RMS_N_READS = 4
static constexpr int	RMS_LOOPED_LIMIT = 4096

Macro Definition Documentation

instantiate_kernel

#define instantiate_kernel	(		name,
			func,
			... )

Value:

  template [[host_name(                     \
      name)]] [[kernel]] decltype(func<__VA_ARGS__>) func<__VA_ARGS__>;

MTL_CONST

#define MTL_CONST

Variable Documentation

MAX_REDUCE_SPECIALIZED_DIMS

constexpr int MAX_REDUCE_SPECIALIZED_DIMS = 4

staticconstexpr

REDUCE_N_READS

constexpr int REDUCE_N_READS = 16

staticconstexpr

RMS_LOOPED_LIMIT

constexpr int RMS_LOOPED_LIMIT = 4096

staticconstexpr

RMS_N_READS

constexpr int RMS_N_READS = 4

staticconstexpr

SOFTMAX_N_READS

constexpr int SOFTMAX_N_READS = 4

staticconstexpr