Reverts changing the MLX_IBV_DEVICES to MLX_JACCL_DEVICES (#2999)

2026-01-14 15:44:17 -08:00
parent 47430159fc
commit 3fe7794f22
3 changed files with 5 additions and 5 deletions
@@ -376,7 +376,7 @@ def launch_jaccl(parser, hosts, args, command):
    env = args.env
    cwd = args.cwd
    env.append(f"MLX_JACCL_COORDINATOR={coordinator}:{args.starting_port}")
-    files = {"MLX_JACCL_DEVICES": json.dumps([h.rdma for h in hosts])}
+    files = {"MLX_IBV_DEVICES": json.dumps([h.rdma for h in hosts])}

    log(args.verbose, "Running", shlex.join(command))