README.md
pyproject.toml
setup.py
csrc/power_attention/api.cpp
csrc/power_attention/src/discumsum_bwd_bf16_sm80.cu
csrc/power_attention/src/discumsum_bwd_fp16_sm80.cu
csrc/power_attention/src/discumsum_bwd_fp32_sm80.cu
csrc/power_attention/src/discumsum_fwd_bf16_sm80.cu
csrc/power_attention/src/discumsum_fwd_fp16_sm80.cu
csrc/power_attention/src/discumsum_fwd_fp32_sm80.cu
csrc/power_attention/src/query_state_bwd_bf16_kq_hdim32_deg2_sm80.cu
csrc/power_attention/src/query_state_bwd_bf16_kq_hdim64_deg2_sm80.cu
csrc/power_attention/src/query_state_bwd_fp16_kq_hdim32_deg2_sm80.cu
csrc/power_attention/src/query_state_bwd_fp16_kq_hdim64_deg2_sm80.cu
csrc/power_attention/src/query_state_fwd_bf16_kq_hdim32_deg2_sm80.cu
csrc/power_attention/src/query_state_fwd_bf16_kq_hdim64_deg2_sm80.cu
csrc/power_attention/src/query_state_fwd_fp16_kq_hdim32_deg2_sm80.cu
csrc/power_attention/src/query_state_fwd_fp16_kq_hdim64_deg2_sm80.cu
csrc/power_attention/src/update_state_bwd_bf16_kq_hdim32_deg2_sm80.cu
csrc/power_attention/src/update_state_bwd_bf16_kq_hdim64_deg2_sm80.cu
csrc/power_attention/src/update_state_bwd_fp16_kq_hdim32_deg2_sm80.cu
csrc/power_attention/src/update_state_bwd_fp16_kq_hdim64_deg2_sm80.cu
csrc/power_attention/src/update_state_fwd_bf16_kq_hdim32_deg2_sm80.cu
csrc/power_attention/src/update_state_fwd_bf16_kq_hdim64_deg2_sm80.cu
csrc/power_attention/src/update_state_fwd_fp16_kq_hdim32_deg2_sm80.cu
csrc/power_attention/src/update_state_fwd_fp16_kq_hdim64_deg2_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim32_bf16_deg1_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim32_bf16_deg2_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim32_bf16_deg3_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim32_bf16_deg4_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim32_fp16_deg1_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim32_fp16_deg2_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim32_fp16_deg3_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim32_fp16_deg4_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim64_bf16_deg1_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim64_bf16_deg2_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim64_bf16_deg3_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim64_bf16_deg4_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim64_fp16_deg1_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim64_fp16_deg2_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim64_fp16_deg3_sm80.cu
csrc/power_attention/src/attention/power_bwd_hdim64_fp16_deg4_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim32_bf16_deg1_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim32_bf16_deg2_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim32_bf16_deg3_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim32_bf16_deg4_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim32_fp16_deg1_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim32_fp16_deg2_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim32_fp16_deg3_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim32_fp16_deg4_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim64_bf16_deg1_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim64_bf16_deg2_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim64_bf16_deg3_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim64_bf16_deg4_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim64_fp16_deg1_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim64_fp16_deg2_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim64_fp16_deg3_causal_sm80.cu
csrc/power_attention/src/attention/power_fwd_hdim64_fp16_deg4_causal_sm80.cu
power_attention_cuda/__init__.py
power_attention_cuda.egg-info/PKG-INFO
power_attention_cuda.egg-info/SOURCES.txt
power_attention_cuda.egg-info/dependency_links.txt
power_attention_cuda.egg-info/requires.txt
power_attention_cuda.egg-info/top_level.txt