[float8nocompile]: Triton kernels for conversion to float8 dtypes for forward pass of Float8LinearNoCompile#1445

Merged

danielvegamyhre merged 3 commits intopytorch:mainfrom danielvegamyhre:fp8kernel1

Dec 19, 2024

+230-1

Commits on Dec 19, 2024

float8nocompile: triton kernel for conversion to fp8e4m3
Daniel Vega-Myhre
committed
float8nocompile: two pass fp8 conversion kernel
Daniel Vega-Myhre
committed
address comments
Daniel Vega-Myhre
committed