log device, dtype

2025-09-22 22:15:44 -04:00
parent 493616fc3d
commit 94cbc6d42d
1 changed files with 1 additions and 0 deletions
--- a/scripts/benchmarks/deepseek_v3_moe.py
+++ b/scripts/benchmarks/deepseek_v3_moe.py
@@ -131,6 +131,7 @@ def benchmark_deepseek_v3(args: argparse.Namespace) -> dict:

    device = resolve_device(args.device)
    dtype = DTYPE_MAP[args.dtype]
+    print(f"device: {device}, dtype: {dtype}")

    if args.n_experts % args.groups != 0:
        raise SystemExit("n-experts must be divisible by groups")