Add Autograd Backward Support for FP4 GEMM Custom Ops - #393

Closed

colalb1 wants to merge 10 commits into

meta-pytorch:mainfrom

colalb1:colalb1/fp4_backprop_kernels

Commits on Jun 16, 2026

Adding backprop functions for FP4-quantized GEMM operations. May need to add these to base PyTorch.
colalb1
committed
725fccb

Commits on Jun 17, 2026

Importing fp4 autograd.
colalb1
committed
03d5bf4
Update fp4 autograd.
colalb1
committed
7c6102c
Autograd tests.
colalb1
committed
633dd2c
Revert.
colalb1
committed
0fbf2ca
Formatting module docustring.
colalb1
committed
7afafc5
Formatting and coments.
colalb1
committed
d1bb78d
Cleanup.
colalb1
committed
9b7bfac

Commits on Jun 23, 2026

Merge branch 'main' into colalb1/fp4_backprop_kernels
colalb1
authored
f7ae28a

Commits on Jul 25, 2026

Merge branch 'main' into colalb1/fp4_backprop_kernels
colalb1
authored
0017d3d