Index A | C | D | E | G | I | J | K | M | N | P | R | S | T | V | X A Accelerator (class in xma.accelerator) AutotuneConfig (class in xma.autotuner.config) C ceil_divide() (in module xma.math) check_power_of_2() (in module xma.math) cpu (Accelerator attribute) cuda (Accelerator attribute) (KernelBackend attribute) D device_count() (Accelerator static method) divide_if_divisible() (in module xma.math) E enable_counters() (in module xma.counters) G get_accelerator() (Accelerator static method) get_cartesian_product_autotune_configs() (in module xma.autotuner.config) get_compatible_accelerator() (KernelBackend method) get_core_count() (Accelerator static method) get_counter_value() (in module xma.counters) get_current_device() (Accelerator static method) get_kernel_backend() (Accelerator static method) get_key_values() (AutotuneConfig method) get_lane_count() (Accelerator static method) get_next_power_of_2() (in module xma.math) get_powers_of_2() (in module xma.math) I increment_counter() (in module xma.counters) is_condition_valid() (AutotuneConfig method) is_counter_enabled() (in module xma.counters) J jax (KernelBackend attribute) K KernelBackend (class in xma.accelerator) M module xma xma.accelerator xma.autotuner.config xma.counters xma.math mps (Accelerator attribute) (KernelBackend attribute) N nki (KernelBackend attribute) P pallas (KernelBackend attribute) R reset_counters() (in module xma.counters) rocm (Accelerator attribute) (KernelBackend attribute) S synchronize() (Accelerator static method) T torch (KernelBackend attribute) tpu (Accelerator attribute) trainium (Accelerator attribute) triton (KernelBackend attribute) V verify_accelerator() (KernelBackend method) X xma module xma.accelerator module xma.autotuner.config module xma.counters module xma.math module