Contents Menu Expand Light mode Dark mode Auto light/dark, in light mode Auto light/dark, in dark mode Skip to content
XMA documentation
XMA documentation

API Reference

  • xma.functional
    • xma.functional.bmm
    • xma.functional.continuous_count
    • xma.functional.cross_entropy
    • xma.functional.fused_residual_add_rmsnorm
    • xma.functional.gru
    • xma.functional.rnn
    • xma.functional.sequence_packing
    • xma.functional.softmax
    • xma.functional.swiglu
    • xma.functional.fused_linear_cross_entropy
    • xma.functional.rmsnorm
    • xma.functional.swiglu_packed
  • xma.layers
    • xma.layers.moe
    • xma.layers.gru
    • xma.layers.rnn

Utilities

  • xma.accelerator
  • xma.counters
Back to top
View this page

xma.counters¶

enable_counters()[source]¶
get_counter_value(key: Any) → int[source]¶
increment_counter(key: Any, value: int = 1) → None[source]¶
is_counter_enabled() → bool[source]¶
reset_counters() → None[source]¶
Previous
xma.accelerator
Copyright © 2026, Mayank Mishra
Made with Sphinx and @pradyunsg's Furo
On this page
  • xma.counters
    • enable_counters()
    • get_counter_value()
    • increment_counter()
    • is_counter_enabled()
    • reset_counters()