tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

log device, dtype

Browse Source
This commit is contained in:
Dan Saunders
2025-09-22 22:15:44 -04:00
parent 493616fc3d
commit 94cbc6d42d
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
scripts/benchmarks/deepseek_v3_moe.py
View File

@@ -131,6 +131,7 @@ def benchmark_deepseek_v3(args: argparse.Namespace) -> dict:
device = resolve_device(args.device) device = resolve_device(args.device)
dtype = DTYPE_MAP[args.dtype] dtype = DTYPE_MAP[args.dtype]
print(f"device: {device}, dtype: {dtype}")
if args.n_experts % args.groups != 0: if args.n_experts % args.groups != 0:
raise SystemExit("n-experts must be divisible by groups") raise SystemExit("n-experts must be divisible by groups")