[Triton] DS fused custom ops#607

Merged

rahulbatra85 merged 43 commits intomainfrom

shaoclee/ds_fused_custom_ops

Jul 16, 2025

Commits on Jul 2, 2025

add fused concat
k50112113
committed
add fused elementwise and pytest
k50112113
committed
clean up
k50112113
committed
add fused quant
k50112113
committed
clean up
k50112113
committed
add fused mxfp4 quant and pytest
k50112113
committed
add gemm_a16w16_atomic and related tests

cagrikymk
authored and
k50112113
committed
extend tests and add config files for bf16 GEMM with atomic add

cagrikymk
authored and
k50112113
committed
fused quant. code cleanup

cagrikymk
authored and
k50112113
committed
formatting changes

cagrikymk
authored and
k50112113
committed
add rms norm quant tests and changes

cagrikymk
authored and
k50112113
committed
update/add DS configs for GEMMs

cagrikymk
authored and
k50112113
committed
add prequant config
k50112113
committed
fix bf16 gemm
k50112113
committed
fix fp4 gemm atomic add
k50112113
committed
black reformatting
k50112113
committed
tune fp4 prequant gemm atomic
k50112113
committed
fix batched fp4 prequant and add new DS configs

cagrikymk
authored and
k50112113
committed
optimize shapes for deepseek

lburzawa
authored and
k50112113
committed
black reformatting
k50112113
committed

Commits on Jul 14, 2025