benchmarks/linear/benchmark_grouped_linear.py · 6d178b4e0465cb6739d7213a6814325aa92d2e38 · OpenDAS / TransformerEngine

"...git@developer.sourcefind.cn:mkm3sc7k5up7/vllm_017.git" did not exist on "fbeb8a6f13f8d47a7a9af6acb968a5cd3810cb24"

[PyTorch][MoE] MXFP8 Support to Reduce CPU Overhead By Fuse Torch Empty Calls (#1934) · 96ee7173

Zhongbo Zhu authored Jul 09, 2025



* functional passed
Signed-off-by: zhongboz <zhongboz@nvidia.com>

* before zero padding in mxfp8 swizzle, use torch zeros to malloc for now
Signed-off-by: zhongboz <zhongboz@nvidia.com>

* format
Signed-off-by: zhongboz <zhongboz@nvidia.com>

* lint
Signed-off-by: zhongboz <zhongboz@nvidia.com>

---------
Signed-off-by: zhongboz <zhongboz@nvidia.com>

96ee7173

benchmark_grouped_linear.py 8.59 KB

Replace benchmark_grouped_linear.py