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

- a -

• align_K : steel_gemm_fused.h
• align_M : steel_gemm_fused.h
• align_N : steel_gemm_fused.h
• arange_kernels : arange.h

- b -

• buf : allocator.h

- c -

• can_convert_from_bfloat : bf16.h
• can_convert_from_complex64 : complex.h
• can_convert_to_bfloat : bf16.h
• can_convert_to_complex64 : complex.h
• copy_kernels : copy.h

- d -

• do_axpby : steel_gemm_fused.h
• do_gather : steel_gemm_fused.h

- e -

• elems_per_thread_ : fft.h

- f -

• f : bf16.h, fp16.h

- g -

• gather_bias : steel_gemm_fused.h
• gather_kernels : indexing.h
• gemv_masked_kernel : gemv_masked.h

- h -

• has_batch : steel_gemm_fused.h

- i -

• inv_ : fft.h
• is_metal_atomic : atomic.h
• is_power_of_2_ : fft.h

- m -

• MAX_REDUCE_SPECIALIZED_DIMS : defines.h

- n -

• next : allocator.h

- o -

• op : binary.h

- p -

• prev : allocator.h

- r -

• rader_11_steps_ : fft.h
• rader_13_steps_ : fft.h
• rader_2_steps_ : fft.h
• rader_3_steps_ : fft.h
• rader_4_steps_ : fft.h
• rader_5_steps_ : fft.h
• rader_6_steps_ : fft.h
• rader_7_steps_ : fft.h
• rader_8_steps_ : fft.h
• rader_m_ : fft.h
• radix_11_steps_ : fft.h
• radix_13_steps_ : fft.h
• radix_2_steps_ : fft.h
• radix_3_steps_ : fft.h
• radix_4_steps_ : fft.h
• radix_5_steps_ : fft.h
• radix_6_steps_ : fft.h
• radix_7_steps_ : fft.h
• radix_8_steps_ : fft.h
• reduce_init_kernels : reduce.h
• reduce_kernels : reduce.h
• REDUCE_N_READS : defines.h
• REDUCE_N_WRITES : defines.h
• reduce_non_atomic_kernels : reduce.h
• RMS_LOOPED_LIMIT : defines.h
• RMS_N_READS : defines.h

- s -

• scan_kernels : scan.h
• scatter_kernels : indexing.h
• simd_size : ops.h
• SIMD_SIZE : quantized.h
• softmax_kernels : softmax.h
• SOFTMAX_N_READS : defines.h
• steel_conv_general_kernels : steel_conv.h
• steel_conv_kernels : steel_conv.h
• steel_gemm_fused_kernels : steel_gemm.h
• steel_gemm_masked_kernels : steel_gemm.h
• steel_gemm_splitk_accum_axbpy_kernels : steel_gemm.h
• steel_gemm_splitk_accum_kernels : steel_gemm.h
• steel_gemm_splitk_kernels : steel_gemm.h

- u -

• u : bf16.h, fp16.h
• use_out_source : steel_gemm_fused.h

- z -

• zero_helper : sort.h