hub / github.com/WaveSpeedAI/QuantumAttention

github.com/WaveSpeedAI/QuantumAttention @main

Chat with this repo

repository ↗ · DeepWiki ↗ · + Follow

79 symbols 249 edges 18 files 1 documented · 1%

What it actually does AI analysis from the code graph — generated when you open this

loading…

README

QuantumAttention

Better (FP8) attention for Hopper

License

This project is licensed under a custom license. Individuals or organizations with an annual revenue of less than 5,000,000 USD are granted permission to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the software under the specified conditions. Schools, educational institutions, and non-profit research purposes are exempt from the revenue restriction and are granted permission to use the software under the same conditions.

For those with an annual revenue equal to or greater than 5,000,000 USD, a separate commercial license is required.

For more details, please refer to the LICENSE file.

Core symbols most depended-on inside this repo

require_dense_memory

called by 7

src/quantum_attn/inductor/kernels/mm_common.py

can_use_attention

called by 5

src/quantum_attn/nn.py

_dynamically_quantize_fp8

called by 4

src/quantum_attn/nn.py

_attention_wrapper

called by 3

src/quantum_attn/nn.py

_fp8_attention_wrapper

called by 3

src/quantum_attn/nn.py

fp8_attention

called by 2

src/quantum_attn/nn.py

_attention_forward

called by 2

src/quantum_attn/ops.py

_fp8_attention_forward

called by 2

src/quantum_attn/ops.py

Shape

Function 76

Class 3

Languages

Python100%

Modules by API surface

src/quantum_attn/nn.py15 symbols

tests/test_interface.py13 symbols

src/quantum_attn/inductor/kernels/attention.py11 symbols

src/quantum_attn/utils/checks.py10 symbols

src/quantum_attn/quantum_attn_interface.py10 symbols

src/quantum_attn/ops.py5 symbols

src/quantum_attn/inductor/kernels/mm_common.py5 symbols

src/quantum_attn/config.py3 symbols

setup.py3 symbols

src/quantum_attn/utils/types.py2 symbols

src/quantum_attn/tk/utils.py1 symbols

src/quantum_attn/tk/attention.py1 symbols

For agents

$ claude mcp add QuantumAttention \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact