From 94cbc6d42d513c912eb86a95d3a0272dd2c5905f Mon Sep 17 00:00:00 2001 From: Dan Saunders Date: Mon, 22 Sep 2025 22:15:44 -0400 Subject: [PATCH] log device, dtype --- scripts/benchmarks/deepseek_v3_moe.py | 1 + 1 file changed, 1 insertion(+) diff --git a/scripts/benchmarks/deepseek_v3_moe.py b/scripts/benchmarks/deepseek_v3_moe.py index 53b79872f..1045810a6 100644 --- 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")