burn_dragon_kernel 0.5.0

Fused GPU kernel crate for burn_dragon execution paths
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

pub use crate::dense_attention::{
    CompiledDenseAttentionPlan, supports_dense_attention_backend,
    try_fused_dense_row_l1_attention_wgpu, try_fused_dense_row_l1_attention_wgpu_with_plan,
};
pub use crate::dense_causal_attention::{
    CompiledDenseCausalAttentionPlan, supports_dense_causal_attention_backend,
    try_fused_dense_causal_attention_wgpu, try_fused_dense_causal_attention_wgpu_with_plan,
};
pub use crate::dense_scores::{
    CompiledDenseScoresPlan, supports_dense_scores_backend, try_fused_dense_row_l1_scores_wgpu,
    try_fused_dense_row_l1_scores_wgpu_with_plan,
};
pub use crate::recurrent::{
    CompiledRecurrentAttentionPlan, RecurrentAttentionOutput, RecurrentProfileSnapshot,
    recurrent_profile_reset, recurrent_profile_snapshot,
    supports_backend as supports_recurrent_backend, try_fused_recurrent_attention_wgpu,
    try_fused_recurrent_attention_wgpu_with_plan,
};