Here is a list of all macros with links to the files they belong to:

- _ -

• __MLX_BFLOAT_NAN__ : bf16.h
• __MLX_HALF_NAN__ : fp16.h
• _MPS_PRIVATE_CLS : gemm.h
• _MPS_PRIVATE_SEL : gemm.h

- a -

• ADD_HALF_BINOPS : half_types.h

- b -

• bfloat16_to_uint16 : bf16_math.h
• bfloat_binop : bf16.h
• bfloat_binop_base : bf16.h
• bfloat_binop_helper : bf16.h
• bfloat_bitop : bf16.h
• bfloat_compop : bf16.h
• bfloat_inplace_bitop : bf16.h
• bfloat_inplace_op : bf16.h
• bfloat_inplace_op_addr_space_helper : bf16.h
• bfloat_inplace_op_helper : bf16.h

- c -

• complex_binop : complex.h
• complex_binop_helper : complex.h

- d -

• DEFINE_DEFAULT_IS_EQUIVALENT : primitives.h
• DEFINE_GRADS : primitives.h
• DEFINE_INPUT_OUTPUT_SHAPE : primitives.h
• DEFINE_PRINT : primitives.h
• DEFINE_VMAP : primitives.h

- f -

• fp16_bf16_binop_helper : half_types.h

- h -

• half_binop : fp16.h
• half_binop_base : fp16.h
• half_binop_helper : fp16.h
• half_bitop : fp16.h
• half_compop : fp16.h
• half_inplace_bitop : fp16.h
• half_inplace_op : fp16.h

- i -

• instantiate_default_limit : utils.h
• instantiate_float_limit : utils.h
• instantiate_kernel : defines.h
• instantiate_metal_math_funcs : bf16_math.h
• instantiate_metal_simd_comm_funcs : bf16_math.h
• instantiate_metal_simd_reduction_funcs : bf16_math.h

- m -

• MAX_OUTPUT_SIZE : fft.h
• MAX_RADIX : fft.h, readwrite.h
• MLX_LAPACK_FUNC : lapack_helper.h
• MLX_MTL_CONST : quantized.h, sort.h
• MLX_MTL_LOOP_UNROLL : sort.h
• MLX_MTL_PRAGMA_UNROLL : utils.h
• MTL_CONST : defines.h

- p -

• POCKETFFT_CACHE_SIZE : pocketfft.h
• POCKETFFT_NO_VECTORS : pocketfft.h
• POCKETFFT_NOINLINE : pocketfft.h
• POCKETFFT_PARTSTEP11 : pocketfft.h
• POCKETFFT_PARTSTEP11a : pocketfft.h
• POCKETFFT_PARTSTEP11a0 : pocketfft.h
• POCKETFFT_PARTSTEP3a : pocketfft.h
• POCKETFFT_PARTSTEP3b : pocketfft.h
• POCKETFFT_PARTSTEP5a : pocketfft.h
• POCKETFFT_PARTSTEP5b : pocketfft.h
• POCKETFFT_PARTSTEP7 : pocketfft.h
• POCKETFFT_PARTSTEP7a : pocketfft.h
• POCKETFFT_PARTSTEP7a0 : pocketfft.h
• POCKETFFT_PREP11 : pocketfft.h
• POCKETFFT_PREP3 : pocketfft.h
• POCKETFFT_PREP5 : pocketfft.h
• POCKETFFT_PREP7 : pocketfft.h
• POCKETFFT_REARRANGE : pocketfft.h
• POCKETFFT_RESTRICT : pocketfft.h

- r -

• RADIX_STEP : fft.h

- s -

• STEEL_CONST : defines.h
• STEEL_PRAGMA_UNROLL : defines.h

- u -

• uint16_to_bfloat16 : bf16_math.h