|
Up
|
|
|
|
|
argmax.wgsl
|
|
|
|
|
argsort.wgsl
|
|
|
|
|
argsort_merge.wgsl
|
|
|
|
|
binary.wgsl
|
|
|
|
|
common_decls.tmpl
|
|
|
|
|
concat.wgsl
|
|
|
|
|
cpy.tmpl.wgsl
|
|
|
|
|
cumsum.wgsl
|
|
|
|
|
embed_wgsl.py
|
|
|
|
|
flash_attn.wgsl
|
|
|
|
|
gated_delta_net.wgsl
|
|
|
|
|
get_rows.wgsl
|
|
|
|
|
glu.tmpl.wgsl
|
|
|
|
|
memset.wgsl
|
|
|
|
|
mul_mat.wgsl
|
|
|
|
|
mul_mat_decls.tmpl
|
|
|
|
|
mul_mat_reg_tile.wgsl
|
|
|
|
|
mul_mat_subgroup_matrix.wgsl
|
|
|
|
|
mul_mat_vec.wgsl
|
|
|
|
|
pad.wgsl
|
|
|
|
|
repeat.wgsl
|
|
|
|
|
rms_norm.wgsl
|
|
|
|
|
rope.tmpl.wgsl
|
|
|
|
|
row_norm.wgsl
|
|
|
|
|
scale.wgsl
|
|
|
|
|
set.wgsl
|
|
|
|
|
set_rows.wgsl
|
|
|
|
|
soft_max.tmpl.wgsl
|
|
|
|
|
solve_tri.wgsl
|
|
|
|
|
ssm_conv.wgsl
|
|
|
|
|
sum_rows.wgsl
|
|
|
|
|
unary.wgsl
|
|
|
|