mirror of
https://git.datalinker.icu/deepseek-ai/DeepSeek-V3.git
synced 2026-06-11 22:33:35 +08:00
Merge 1ea5a1a7ed7429e25ba2fa2c76fe74fa5968d348 into 9b4e9788e4a3a731f7567338ed15d3ec549ce03b
This commit is contained in:
commit
766aa14e40
@ -10,7 +10,7 @@ import torch.distributed as dist
|
|||||||
from kernel import act_quant, weight_dequant, fp8_gemm
|
from kernel import act_quant, weight_dequant, fp8_gemm
|
||||||
|
|
||||||
|
|
||||||
world_size = 1
|
world_size = 12
|
||||||
rank = 0
|
rank = 0
|
||||||
block_size = 128
|
block_size = 128
|
||||||
gemm_impl: Literal["bf16", "fp8"] = "bf16"
|
gemm_impl: Literal["bf16", "fp8"] = "bf16"
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user