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 -

• IDX_ARG_0 : indexing.h
• IDX_ARG_1 : indexing.h
• IDX_ARG_10 : indexing.h
• IDX_ARG_2 : indexing.h
• IDX_ARG_3 : indexing.h
• IDX_ARG_4 : indexing.h
• IDX_ARG_5 : indexing.h
• IDX_ARG_6 : indexing.h
• IDX_ARG_7 : indexing.h
• IDX_ARG_8 : indexing.h
• IDX_ARG_9 : indexing.h
• IDX_ARG_N : indexing.h
• IDX_ARR_0 : indexing.h
• IDX_ARR_1 : indexing.h
• IDX_ARR_10 : indexing.h
• IDX_ARR_2 : indexing.h
• IDX_ARR_3 : indexing.h
• IDX_ARR_4 : indexing.h
• IDX_ARR_5 : indexing.h
• IDX_ARR_6 : indexing.h
• IDX_ARR_7 : indexing.h
• IDX_ARR_8 : indexing.h
• IDX_ARR_9 : indexing.h
• IDX_ARR_N : indexing.h
• instantiate_default_limit : utils.h
• instantiate_float_limit : utils.h
• instantiate_metal_math_funcs : bf16_math.h
• instantiate_metal_simd_comm_funcs : bf16_math.h
• instantiate_metal_simd_reduction_funcs : bf16_math.h
• instantiate_reduce_from_types : reduce_inst.h
• instantiate_reduce_from_types_helper : reduce_inst.h
• instantiate_reduce_helper_64b : reduce_inst.h
• instantiate_reduce_helper_floats : reduce_inst.h
• instantiate_reduce_helper_ints : reduce_inst.h
• instantiate_reduce_helper_types : reduce_inst.h
• instantiate_reduce_helper_uints : reduce_inst.h
• instantiate_reduce_ops : reduce_inst.h

- m -

• MLX_LAPACK_FUNC : lapack_helper.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

- s -

• STEEL_CONST : utils.h
• STEEL_PRAGMA_UNROLL : utils.h

- u -

• uint16_to_bfloat16 : bf16_math.h