Merge 1ea5a1a7ed7429e25ba2fa2c76fe74fa5968d348 into 9b4e9788e4a3a731f7567338ed15d3ec549ce03b

This commit is contained in:
Kunal Janjirala 2025-08-29 08:27:17 +03:00 committed by GitHub
commit 766aa14e40
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -10,7 +10,7 @@ import torch.distributed as dist
from kernel import act_quant, weight_dequant, fp8_gemm
world_size = 1
world_size = 12
rank = 0
block_size = 128
gemm_impl: Literal["bf16", "fp8"] = "bf16"